瀏覽代碼

Update requirements.txt (#2021)

* Update requirements.txt

* Update ci-testing.yml

* Update hubconf.py
5.0
Glenn Jocher GitHub 3 年之前
父節點
當前提交
9a3da79b4a
沒有發現已知的金鑰在資料庫的簽署中 GPG Key ID: 4AEE18F83AFDEB23
共有 3 個文件被更改,包括 3 次插入2 次删除
  1. +1
    -0
      .github/workflows/ci-testing.yml
  2. +1
    -1
      hubconf.py
  3. +1
    -1
      requirements.txt

+ 1
- 0
.github/workflows/ci-testing.yml 查看文件

@@ -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

+ 1
- 1
hubconf.py 查看文件

@@ -137,5 +137,5 @@ if __name__ == '__main__':

imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
results = model(imgs)
results.show()
results.print()
results.save()

+ 1
- 1
requirements.txt 查看文件

@@ -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

Loading…
取消
儲存