Skip to content

Commit

Permalink
Update requirements.txt (ultralytics#2021)
Browse files Browse the repository at this point in the history
* Update requirements.txt

* Update ci-testing.yml

* Update hubconf.py
  • Loading branch information
glenn-jocher committed Jan 23, 2021
1 parent 55f7a18 commit c6c438c
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
1 change: 1 addition & 0 deletions .github/workflows/ci-testing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 runs/train/exp/weights/last.pt --device $di
python hubconf.py # hub
python models/yolo.py --cfg models/${{ matrix.model }}.yaml # inspect
python models/export.py --img 128 --batch 1 --weights weights/${{ matrix.model }}.pt # export
shell: bash
2 changes: 1 addition & 1 deletion hubconf.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,5 +137,5 @@ def custom(path_or_model='path/to/model.pt', autoshape=True):

imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
results = model(imgs)
results.show()
results.print()
results.save()
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow
PyYAML>=5.3
PyYAML>=3.13
scipy>=1.4.1
tensorboard>=2.2
torch>=1.7.0
Expand Down

0 comments on commit c6c438c

Please sign in to comment.