zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update requirements.txt (#2021) 

* Update requirements.txt

* Update ci-testing.yml

* Update hubconf.py
This commit is contained in:
Glenn Jocher 2021-01-23 12:51:04 -08:00 committed by GitHub
parent 2fc4760257
commit 9a3da79b4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.github/workflows/ci-testing.yml vendored
View File

@ -74,6 +74,7 @@ jobs:
python test.py --img 128 --batch 16 --weights weights/${{ matrix.model }}.pt --device $di python test.py --img 128 --batch 16 --weights weights/${{ matrix.model }}.pt --device $di
python test.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di python test.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di
python hubconf.py # hub
python models/yolo.py --cfg models/${{ matrix.model }}.yaml # inspect python models/yolo.py --cfg models/${{ matrix.model }}.yaml # inspect
python models/export.py --img 128 --batch 1 --weights weights/${{ matrix.model }}.pt # export python models/export.py --img 128 --batch 1 --weights weights/${{ matrix.model }}.pt # export
shell: bash shell: bash

2
hubconf.py
View File

@ -137,5 +137,5 @@ if __name__ == '__main__':
imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')] imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
results = model(imgs) results = model(imgs)
results.show()
results.print() results.print()
results.save()

2
requirements.txt
View File

@ -6,7 +6,7 @@ matplotlib>=3.2.2
numpy>=1.18.5 numpy>=1.18.5
opencv-python>=4.1.2 opencv-python>=4.1.2
Pillow Pillow
PyYAML>=5.3 PyYAML>=3.13
scipy>=1.4.1 scipy>=1.4.1
tensorboard>=2.2 tensorboard>=2.2
torch>=1.7.0 torch>=1.7.0