Unverified 提交 94705a95 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Add Hub custom models to CI tests (#4978)

* Update ci-testing.yml for Hub custom model tests * Update ci-testing.yml
上级 0c874787
...@@ -4,10 +4,10 @@ name: CI CPU testing ...@@ -4,10 +4,10 @@ name: CI CPU testing
on: # https://help.github.com/en/actions/reference/events-that-trigger-workflows on: # https://help.github.com/en/actions/reference/events-that-trigger-workflows
push: push:
branches: [master] branches: [ master ]
pull_request: pull_request:
# The branches below must be a subset of the branches above # The branches below must be a subset of the branches above
branches: [master] branches: [ master ]
schedule: schedule:
- cron: '0 0 * * *' # Runs at 00:00 UTC every day - cron: '0 0 * * *' # Runs at 00:00 UTC every day
...@@ -18,9 +18,9 @@ jobs: ...@@ -18,9 +18,9 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
os: [ubuntu-latest, macos-latest, windows-latest] os: [ ubuntu-latest, macos-latest, windows-latest ]
python-version: [3.8] python-version: [ 3.8 ]
model: ['yolov5s'] # models to test model: [ 'yolov5s' ] # models to test
# Timeout: https://stackoverflow.com/a/59076067/4521646 # Timeout: https://stackoverflow.com/a/59076067/4521646
timeout-minutes: 50 timeout-minutes: 50
...@@ -65,19 +65,25 @@ jobs: ...@@ -65,19 +65,25 @@ jobs:
- name: Tests workflow - name: Tests workflow
run: | run: |
# export PYTHONPATH="$PWD" # to run '$ python *.py' files in subdirectories # export PYTHONPATH="$PWD" # to run '$ python *.py' files in subdirectories
di=cpu # inference devices # define device di=cpu # device
# train # Train
python train.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di python train.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di
# detect # Val
python detect.py --weights ${{ matrix.model }}.pt --device $di
python detect.py --weights runs/train/exp/weights/last.pt --device $di
# val
python val.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --device $di python val.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --device $di
python val.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di python val.py --img 128 --batch 16 --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 python hubconf.py # hub
# Export
python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model
python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow model python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow model
python export.py --img 128 --batch 1 --weights ${{ matrix.model }}.pt --include torchscript onnx # export python export.py --img 128 --batch 1 --weights ${{ matrix.model }}.pt --include torchscript onnx # export
# Python
python - <<EOF
import torch
model = torch.hub.load('ultralytics/yolov5', 'custom', path='runs/train/exp/weights/last.pt')
EOF
shell: bash shell: bash
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论