diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index 1210400e..b0a6bfd9 100644 --- a/.github/workflows/ci-testing.yml +++ b/.github/workflows/ci-testing.yml @@ -1,93 +1,99 @@ # YOLOv3 🚀 by Ultralytics, GPL-3.0 license +# YOLOv3 Continuous Integration (CI) GitHub Actions tests -name: CI CPU testing +name: YOLOv3 CI -on: # https://help.github.com/en/actions/reference/events-that-trigger-workflows +on: push: branches: [ master ] pull_request: - # The branches below must be a subset of the branches above branches: [ master ] schedule: - - cron: '0 0 * * *' # Runs at 00:00 UTC every day + - cron: '0 0 * * *' # runs at 00:00 UTC every day jobs: - cpu-tests: - + Tests: + timeout-minutes: 60 runs-on: ${{ matrix.os }} strategy: fail-fast: false matrix: - os: [ ubuntu-latest, windows-latest ] - python-version: [ 3.9 ] - model: [ 'yolov3-tiny' ] # models to test - - # Timeout: https://stackoverflow.com/a/59076067/4521646 - timeout-minutes: 50 + os: [ ubuntu-latest, windows-latest ] # macos-latest bug https://github.com/ultralytics/yolov5/pull/9049 + python-version: [ '3.10' ] + model: [ yolov3-tiny ] + include: + - os: ubuntu-latest + python-version: '3.7' # '3.6.8' min + model: yolov3-tiny + - os: ubuntu-latest + python-version: '3.8' + model: yolov3-tiny + - os: ubuntu-latest + python-version: '3.9' + model: yolov3-tiny + - os: ubuntu-latest + python-version: '3.8' # torch 1.7.0 requires python >=3.6, <=3.8 + model: yolov3-tiny + torch: '1.7.0' # min torch version CI https://pypi.org/project/torchvision/ steps: - uses: actions/checkout@v3 - - name: Set up Python ${{ matrix.python-version }} - uses: actions/setup-python@v4 + - uses: actions/setup-python@v4 with: python-version: ${{ matrix.python-version }} - - # Note: This uses an internal pip API and may not always work - # https://github.com/actions/cache/blob/master/examples.md#multiple-oss-in-a-workflow - - name: Get pip cache + - name: Get cache dir + # https://github.com/actions/cache/blob/master/examples.md#multiple-oss-in-a-workflow id: pip-cache - run: | - python -c "from pip._internal.locations import USER_CACHE_DIR; print('::set-output name=dir::' + USER_CACHE_DIR)" - + run: echo "::set-output name=dir::$(pip cache dir)" - name: Cache pip uses: actions/cache@v3 with: path: ${{ steps.pip-cache.outputs.dir }} key: ${{ runner.os }}-${{ matrix.python-version }}-pip-${{ hashFiles('requirements.txt') }} - restore-keys: | - ${{ runner.os }}-${{ matrix.python-version }}-pip- - - # Known Keras 2.7.0 issue: https://github.com/ultralytics/yolov5/pull/5486 - - name: Install dependencies + restore-keys: ${{ runner.os }}-${{ matrix.python-version }}-pip- + - name: Install requirements run: | - python -m pip install --upgrade pip - pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html + python -m pip install --upgrade pip wheel + if [ "${{ matrix.torch }}" == "1.7.0" ]; then + pip install -r requirements.txt torch==1.7.0 torchvision==0.8.1 --extra-index-url https://download.pytorch.org/whl/cpu + else + pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu + fi + shell: bash # for Windows compatibility + - name: Check environment + run: | + python -c "import utils; utils.notebook_init()" + echo "RUNNER_OS is ${{ runner.os }}" + echo "GITHUB_EVENT_NAME is ${{ github.event_name }}" + echo "GITHUB_WORKFLOW is ${{ github.workflow }}" + echo "GITHUB_ACTOR is ${{ github.actor }}" + echo "GITHUB_REPOSITORY is ${{ github.repository }}" + echo "GITHUB_REPOSITORY_OWNER is ${{ github.repository_owner }}" python --version pip --version pip list - shell: bash - - # - name: W&B login - # run: wandb login 345011b3fb26dc8337fd9b20e53857c1d403f2aa - - - name: Download data - run: | - # curl -L -o tmp.zip https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip - # unzip -q tmp.zip -d ../ - # rm tmp.zip - - - name: Tests workflow + - name: Test detection + shell: bash # for Windows compatibility run: | # export PYTHONPATH="$PWD" # to run '$ python *.py' files in subdirectories - di=cpu # device - - # Train - python train.py --img 64 --batch 32 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di - # Val - python val.py --img 64 --batch 32 --weights ${{ matrix.model }}.pt --device $di - python val.py --img 64 --batch 32 --weights runs/train/exp/weights/last.pt --device $di - # Detect - python detect.py --weights ${{ matrix.model }}.pt --device $di - python detect.py --weights runs/train/exp/weights/last.pt --device $di - python hubconf.py # hub - # Export - python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model - # python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow model (YOLOv3 not supported) - python export.py --img 64 --batch 1 --weights runs/train/exp/weights/last.pt --include torchscript # export - # Python + m=${{ matrix.model }} # official weights + b=runs/train/exp/weights/best # best.pt checkpoint + python train.py --imgsz 64 --batch 32 --weights $m.pt --cfg $m.yaml --epochs 1 --device cpu # train + for d in cpu; do # devices + for w in $m $b; do # weights + python val.py --imgsz 64 --batch 32 --weights $w.pt --device $d # val + python detect.py --imgsz 64 --weights $w.pt --device $d # detect + done + done + python hubconf.py --model $m # hub + # python models/tf.py --weights $m.pt # build TF model + python models/yolo.py --cfg $m.yaml # build PyTorch model + python export.py --weights $m.pt --img 64 --include torchscript # export python - <