From 4aaf354d8a7f66bb8baa4e5ba73e83b960d8f2d1 Mon Sep 17 00:00:00 2001 From: fcakyon Date: Wed, 25 Aug 2021 12:44:51 +0300 Subject: [PATCH] fix package testing --- .github/workflows/package_testing.yml | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/.github/workflows/package_testing.yml b/.github/workflows/package_testing.yml index 023beb7..82857eb 100644 --- a/.github/workflows/package_testing.yml +++ b/.github/workflows/package_testing.yml @@ -56,21 +56,22 @@ jobs: python -m unittest - name: Test scripts run: | + pip install -e . di=cpu # inference devices # define device # train python yolov5/train.py --img 128 --batch 16 --weights yolov5/weights/yolov5s.pt --cfg models/yolov5s.yaml --epochs 1 --device $di - yolo_train --img 128 --batch 16 --weights yolov5/weights/yolov5s.pt --cfg models/yolov5s.yaml --epochs 1 --device $di + yolov5 train --img 128 --batch 16 --weights yolov5/weights/yolov5s.pt --cfg models/yolov5s.yaml --epochs 1 --device $di # detect python yolov5/detect.py --weights yolov5/weights/yolov5s.pt --device $di - yolo_detect --weights yolov5/weights/yolov5s.pt --device $di + yolov5 detect --weights yolov5/weights/yolov5s.pt --device $di python yolov5/detect.py --weights runs/train/exp/weights/last.pt --device $di - yolo_detect --weights runs/train/exp/weights/last.pt --device $di - # test - python yolov5/test.py --img 128 --batch 16 --weights yolov5/weights/yolov5s.pt --device $di - yolo_test --img 128 --batch 16 --weights yolov5/weights/yolov5s.pt --device $di - python yolov5/test.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di - yolo_test --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di + yolov5 detect --weights runs/train/exp/weights/last.pt --device $di + # val + python yolov5/val.py --img 128 --batch 16 --weights yolov5/weights/yolov5s.pt --device $di + yolov5 val --data yolov5/data/coco128.yaml --img 128 --batch 16 --weights yolov5/weights/yolov5s.pt --device $di + python yolov5/val.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di + yolov5 val --data yolov5/data/coco128.yaml --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di # export pip install onnx coremltools python yolov5/export.py --weights yolov5/weights/yolov5s.pt --device $di - yolo_export --weights yolov5/weights/yolov5s.pt --device $di + yolov5 export --weights yolov5/weights/yolov5s.pt --device $di