diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index 660efe8..659cb49 100644 --- a/.github/workflows/ci-testing.yml +++ b/.github/workflows/ci-testing.yml @@ -47,21 +47,28 @@ jobs: pip list shell: bash - - name: Download data - run: | - python3 -c "from utils.google_utils import *; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')" && mv ./coco128 ../ + #- name: Download data + # run: | + # python3 -c "from utils.google_utils import *; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')" && mv ./coco128 ../ - name: Tests run: | for x in yolov5s yolov5m yolov5l yolov5x # models do di=cpu # inference devices - python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device $di # train - python detect.py --weights $x.pt --device $di # detect official - python detect.py --weights runs/exp0/weights/last.pt --device $di # detect custom - python test.py --weights $x.pt --device $di # test official - python test.py --weights runs/exp0/weights/last.pt --device $di # test custom - python models/yolo.py --cfg $x.yaml # inspect - python models/export.py --weights $x.pt --img 640 --batch 1 # export + # train + python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device $di --batch-size 2 + # detect official + python detect.py --weights $x.pt --device $di --batch-size 2 + # detect custom + python detect.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2 + # test official + python test.py --weights $x.pt --device $di --batch-size 2 + # test custom + python test.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2 + # inspect + python models/yolo.py --cfg $x.yaml + # export + python models/export.py --weights $x.pt --img 640 --batch 1 done shell: bash