glenn-jocher
commited on
Commit
•
9a3da79
1
Parent(s):
2fc4760
Update requirements.txt (#2021)
Browse files* Update requirements.txt
* Update ci-testing.yml
* Update hubconf.py
- .github/workflows/ci-testing.yml +1 -0
- hubconf.py +1 -1
- requirements.txt +1 -1
.github/workflows/ci-testing.yml
CHANGED
@@ -74,6 +74,7 @@ jobs:
|
|
74 |
python test.py --img 128 --batch 16 --weights weights/${{ matrix.model }}.pt --device $di
|
75 |
python test.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di
|
76 |
|
|
|
77 |
python models/yolo.py --cfg models/${{ matrix.model }}.yaml # inspect
|
78 |
python models/export.py --img 128 --batch 1 --weights weights/${{ matrix.model }}.pt # export
|
79 |
shell: bash
|
|
|
74 |
python test.py --img 128 --batch 16 --weights weights/${{ matrix.model }}.pt --device $di
|
75 |
python test.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di
|
76 |
|
77 |
+
python hubconf.py # hub
|
78 |
python models/yolo.py --cfg models/${{ matrix.model }}.yaml # inspect
|
79 |
python models/export.py --img 128 --batch 1 --weights weights/${{ matrix.model }}.pt # export
|
80 |
shell: bash
|
hubconf.py
CHANGED
@@ -137,5 +137,5 @@ if __name__ == '__main__':
|
|
137 |
|
138 |
imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
|
139 |
results = model(imgs)
|
140 |
-
results.show()
|
141 |
results.print()
|
|
|
|
137 |
|
138 |
imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
|
139 |
results = model(imgs)
|
|
|
140 |
results.print()
|
141 |
+
results.save()
|
requirements.txt
CHANGED
@@ -6,7 +6,7 @@ matplotlib>=3.2.2
|
|
6 |
numpy>=1.18.5
|
7 |
opencv-python>=4.1.2
|
8 |
Pillow
|
9 |
-
PyYAML>=
|
10 |
scipy>=1.4.1
|
11 |
tensorboard>=2.2
|
12 |
torch>=1.7.0
|
|
|
6 |
numpy>=1.18.5
|
7 |
opencv-python>=4.1.2
|
8 |
Pillow
|
9 |
+
PyYAML>=3.13
|
10 |
scipy>=1.4.1
|
11 |
tensorboard>=2.2
|
12 |
torch>=1.7.0
|