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 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>=5.3
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