From 9a3da79b4a54217f145ca29ffd8641c98ce1613d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 23 Jan 2021 12:51:04 -0800 Subject: [PATCH] Update requirements.txt (#2021) * Update requirements.txt * Update ci-testing.yml * Update hubconf.py --- .github/workflows/ci-testing.yml | 1 + hubconf.py | 2 +- requirements.txt | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index 5999ee6a9055..df508474a955 100644 --- a/.github/workflows/ci-testing.yml +++ b/.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 diff --git a/hubconf.py b/hubconf.py index c4485a42e335..2a34813310e8 100644 --- a/hubconf.py +++ b/hubconf.py @@ -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() diff --git a/requirements.txt b/requirements.txt index 3c23f2b750a2..2b447aef4d39 100755 --- a/requirements.txt +++ b/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