remove irrelevant benchmark ci
parent
cf28bbbaec
commit
e29d794d21
|
@ -12,40 +12,6 @@ on:
|
||||||
- cron: '0 0 * * *' # runs at 00:00 UTC every day
|
- cron: '0 0 * * *' # runs at 00:00 UTC every day
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
Benchmarks:
|
|
||||||
runs-on: ${{ matrix.os }}
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
os: [ ubuntu-latest ]
|
|
||||||
python-version: [ '3.10' ] # requires python<=3.10
|
|
||||||
model: [ yolov5n ]
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v3
|
|
||||||
- uses: actions/setup-python@v4
|
|
||||||
with:
|
|
||||||
python-version: ${{ matrix.python-version }}
|
|
||||||
cache: 'pip' # caching pip dependencies
|
|
||||||
- name: Install requirements
|
|
||||||
run: |
|
|
||||||
python -m pip install --upgrade pip wheel
|
|
||||||
pip install -r requirements.txt coremltools openvino-dev tensorflow-cpu --extra-index-url https://download.pytorch.org/whl/cpu
|
|
||||||
python --version
|
|
||||||
pip --version
|
|
||||||
pip list
|
|
||||||
- name: Benchmark DetectionModel
|
|
||||||
run: |
|
|
||||||
python benchmarks.py --data coco128.yaml --weights ${{ matrix.model }}.pt --img 320 --hard-fail 0.29
|
|
||||||
- name: Benchmark SegmentationModel
|
|
||||||
run: |
|
|
||||||
python benchmarks.py --data coco128-seg.yaml --weights ${{ matrix.model }}-seg.pt --img 320 --hard-fail 0.22
|
|
||||||
- name: Test predictions
|
|
||||||
run: |
|
|
||||||
python export.py --weights ${{ matrix.model }}-cls.pt --include onnx --img 224
|
|
||||||
python detect.py --weights ${{ matrix.model }}.onnx --img 320
|
|
||||||
python segment/predict.py --weights ${{ matrix.model }}-seg.onnx --img 320
|
|
||||||
python classify/predict.py --weights ${{ matrix.model }}-cls.onnx --img 224
|
|
||||||
|
|
||||||
Tests:
|
Tests:
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
runs-on: ${{ matrix.os }}
|
runs-on: ${{ matrix.os }}
|
||||||
|
|
Loading…
Reference in New Issue