From 9aa2efdbe85f947a643535810d6f36f6c53d4653 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 22 Jan 2021 16:02:25 -0800 Subject: [PATCH 1/3] Update requirements.txt --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 From f84f98264ac7f75e973c85e9665e21ed19e48d4e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 22 Jan 2021 16:37:28 -0800 Subject: [PATCH 2/3] Update ci-testing.yml --- .github/workflows/ci-testing.yml | 1 + 1 file changed, 1 insertion(+) 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 From 39a646963574ad8f5730e27f0d7d1cca3def8b44 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 23 Jan 2021 12:41:57 -0800 Subject: [PATCH 3/3] Update hubconf.py --- hubconf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()