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