update yolov5 version to 6.0.1 in ci tests (#245)

* update yolov5 version to 6.0.1 in ci tests * update yolov5 version to 6.0.1 in package tests
obss · Oct 19, 2021 · d25ee10 · d25ee10
1 parent 53c5d7c
commit d25ee10
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -67,9 +67,9 @@ jobs:
         if: matrix.operating-system == 'macos-latest'
         run: pip install torch==1.8.1 torchvision==0.9.1
 
-      - name: Install MMDetection(2.16.0), YOLOv5(5.0.9) and Norfair(0.3.1)
+      - name: Install MMDetection(2.16.0), YOLOv5(6.0.1) and Norfair(0.3.1)
         run: >
-          pip install mmcv-full==1.3.12 mmdet==2.16.0 yolov5==5.0.9 norfair==0.3.1
+          pip install mmcv-full==1.3.12 mmdet==2.16.0 yolov5==6.0.1 norfair==0.3.1
 
       - name: Lint with flake8, black and isort
         run: |
@@ -90,7 +90,7 @@ jobs:
         run: >
           pip install -e .
 
-      - name: Test SAHI CLI for MMDetection(2.16.0), YOLOv5(5.0.9) and Norfair(0.3.1)
+      - name: Test SAHI CLI for MMDetection(2.16.0), YOLOv5(6.0.1) and Norfair(0.3.1)
         run: |
           # help
           sahi --help

diff --git a/.github/workflows/package_testing.yml b/.github/workflows/package_testing.yml
@@ -66,16 +66,16 @@ jobs:
         if: matrix.operating-system == 'macos-latest'
         run: pip install torch==1.8.1 torchvision==0.9.1
 
-      - name: Install MMDetection(2.16.0), YOLOv5(5.0.9) and Norfair(0.3.1)
+      - name: Install MMDetection(2.16.0), YOLOv5(6.0.1) and Norfair(0.3.1)
         run: >
-          pip install mmcv-full==1.3.12 mmdet==2.16.0 yolov5==5.0.9 norfair==0.3.1
+          pip install mmcv-full==1.3.12 mmdet==2.16.0 yolov5==6.0.1 norfair==0.3.1
 
       - name: Test with unittest
         run: |
           pip install pytest
           python -m unittest
 
-      - name: Test SAHI CLI for MMDetection(2.16.0), YOLOv5(5.0.9) and Norfair(0.3.1)
+      - name: Test SAHI CLI for MMDetection(2.16.0), YOLOv5(6.0.1) and Norfair(0.3.1)
         run: |
           # predict mmdet
           sahi predict --source tests/data/ --model_path tests/data/models/mmdet_retinanet/retinanet_r50_fpn_2x_coco_20200131-fdb43119.pth --model_config_path tests/data/models/mmdet_retinanet/retinanet_r50_fpn_1x_coco.py --image_size 320