fix package testing

fcakyon · Aug 25, 2021 · 4aaf354 · 4aaf354
1 parent c9631c9
commit 4aaf354
Showing 1 changed file with 10 additions and 9 deletions.
diff --git 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