Package Testing #926
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Package Testing | |
| on: | |
| schedule: | |
| - cron: '0 0 * * *' # Runs at 00:00 UTC every day | |
| jobs: | |
| build: | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| operating-system: [ubuntu-latest, windows-latest, macos-latest] | |
| python-version: [3.7, 3.8, 3.9] | |
| fail-fast: false | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v3 | |
| - name: Set up Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Restore Ubuntu cache | |
| uses: actions/cache@v3 | |
| if: matrix.operating-system == 'ubuntu-latest' | |
| with: | |
| path: ~/.cache/pip | |
| key: ${{ matrix.os }}-${{ matrix.python-version }}-${{ hashFiles('**/setup.py')}} | |
| restore-keys: ${{ matrix.os }}-${{ matrix.python-version }}- | |
| - name: Restore MacOS cache | |
| uses: actions/cache@v3 | |
| if: matrix.operating-system == 'macos-latest' | |
| with: | |
| path: ~/Library/Caches/pip | |
| key: ${{ matrix.os }}-${{ matrix.python-version }}-${{ hashFiles('**/setup.py')}} | |
| restore-keys: ${{ matrix.os }}-${{ matrix.python-version }}- | |
| - name: Restore Windows cache | |
| uses: actions/cache@v3 | |
| if: matrix.operating-system == 'windows-latest' | |
| with: | |
| path: ~\AppData\Local\pip\Cache | |
| key: ${{ matrix.os }}-${{ matrix.python-version }}-${{ hashFiles('**/setup.py')}} | |
| restore-keys: ${{ matrix.os }}-${{ matrix.python-version }}- | |
| - name: Update pip | |
| run: python -m pip install --upgrade pip | |
| - name: Install PyTorch(1.10.2) and TorchVision(0.11.3) on Linux and Windows | |
| if: > | |
| matrix.operating-system == 'ubuntu-latest' || | |
| matrix.operating-system == 'windows-latest' | |
| run: > | |
| pip install torch==1.10.2+cpu torchvision==0.11.3+cpu | |
| -f https://download.pytorch.org/whl/torch_stable.html | |
| - name: Install PyTorch(1.10.1) and TorchVision(0.11.2) on MacOS | |
| if: matrix.operating-system == 'macos-latest' | |
| run: pip install torch==1.10.1 torchvision==0.11.2 | |
| - name: Install MMDetection(2.28.1) with MMCV(1.7.0) | |
| run: > | |
| pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html | |
| pip install mmdet==2.28.1 | |
| - name: Install YOLOv5(7.0.9) | |
| run: > | |
| pip install yolov5==7.0.9 | |
| - name: Install DeepSparse | |
| run: > | |
| pip install deepsparse | |
| - name: Install Detectron2(0.6) | |
| run: > | |
| python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.10/index.html | |
| - name: Install Transformers(4.25.1) | |
| run: > | |
| pip install transformers==4.25.1 | |
| - name: Install pycocotools(2.0.6) | |
| run: > | |
| pip install pycocotools==2.0.6 | |
| - name: Install ultralytics | |
| run: > | |
| pip install ultralytics==8.0.99 | |
| - name: Install super-gradients | |
| run: > | |
| pip install super-gradients==3.1.2 | |
| - name: Install latest SAHI package | |
| run: > | |
| pip install --upgrade --force-reinstall sahi | |
| - name: Unittest for SAHI+YOLOV5/MMDET/Detectron2 on all platforms | |
| run: | | |
| python -m unittest | |
| - name: Test SAHI CLI | |
| run: | | |
| # predict mmdet | |
| sahi predict --source tests/data/ --novisual --model_path tests/data/models/mmdet_yolox/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth --model_config_path tests/data/models/mmdet_retinanet/retinanet_r50_fpn_1x_coco.py --image_size 320 | |
| sahi predict --source tests/data/coco_utils/terrain1.jpg --export_pickle --export_crop --model_path tests/data/models/mmdet_yolox/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth --model_config_path tests/data/models/mmdet_yolox/yolox_tiny_8x8_300e_coco.py --image_size 320 | |
| sahi predict --source tests/data/coco_utils/ --novisual --dataset_json_path tests/data/coco_utils/combined_coco.json --model_path tests/data/models/mmdet_yolox/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth --model_config_path tests/data/models/mmdet_yolox/yolox_tiny_8x8_300e_coco.py --image_size 320 | |
| # predict yolov5 | |
| sahi predict --no_sliced_prediction --model_type yolov5 --source tests/data/coco_utils/terrain1.jpg --novisual --model_path tests/data/models/yolov5/yolov5s6.pt --image_size 320 | |
| sahi predict --model_type yolov5 --source tests/data/ --novisual --model_path tests/data/models/yolov5/yolov5s6.pt --image_size 320 | |
| sahi predict --model_type yolov5 --source tests/data/coco_utils/terrain1.jpg --export_pickle --export_crop --model_path tests/data/models/yolov5/yolov5s6.pt --image_size 320 | |
| sahi predict --model_type yolov5 --source tests/data/coco_utils/ --novisual --dataset_json_path tests/data/coco_utils/combined_coco.json --model_path tests/data/models/yolov5/yolov5s6.pt --image_size 320 | |
| # coco yolov5 | |
| sahi coco yolov5 --image_dir tests/data/coco_utils/ --dataset_json_path tests/data/coco_utils/combined_coco.json --train_split 0.9 | |
| # coco evaluate | |
| sahi coco evaluate --dataset_json_path tests/data/coco_evaluate/dataset.json --result_json_path tests/data/coco_evaluate/result.json | |
| # coco analyse | |
| sahi coco analyse --dataset_json_path tests/data/coco_evaluate/dataset.json --result_json_path tests/data/coco_evaluate/result.json --out_dir tests/data/coco_evaluate/ |