Skip to content

Commit

Permalink
Merge pull request #1639 from mikel-brostrom/download-det-n-emb-bench…
Browse files Browse the repository at this point in the history
…mark

download dets n embs from release
  • Loading branch information
mikel-brostrom authored Sep 18, 2024
2 parents 517d4d1 + ae8141f commit 115e207
Showing 1 changed file with 6 additions and 45 deletions.
51 changes: 6 additions & 45 deletions .github/workflows/benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,46 +16,8 @@ on:

jobs:

generate-detections-embeddings:
runs-on: ubuntu-latest
timeout-minutes: 50
steps:
- uses: actions/checkout@v4

- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: '3.9'
cache: 'pip'

- name: Install requirements
run: |
if [[ "$OSTYPE" == "darwin"* ]]; then
# macOS
sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
# Linux
sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
fi
sudo apt-get install -y jq
python -m pip install --upgrade pip setuptools wheel poetry
poetry config virtualenvs.create false
poetry lock --no-update
poetry install --with yolo
- name: Generate detections and embeddings
run: |
python tracking/val.py generate_dets_embs --source ./assets/MOT17-mini/train --yolo-model yolov10n.pt --reid-model osnet_x0_25_msmt17.pt --imgsz 320
- name: Upload Detections and Embeddings
uses: actions/upload-artifact@v3
with:
name: run-folder
path: runs/

mot-metrics-benchmark:
runs-on: ${{ matrix.os }}
needs: generate-detections-embeddings
strategy:
fail-fast: false
matrix:
Expand All @@ -82,22 +44,21 @@ jobs:
# Linux
sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
fi
sudo apt-get install -y jq
sudo apt-get install -y jq unzip
python -m pip install --upgrade pip setuptools wheel poetry
poetry config virtualenvs.create false
poetry lock --no-update
poetry install --with yolo
- name: Download Detections and Embeddings
uses: actions/download-artifact@v3
with:
name: run-folder
path: runs/
- name: Download run.zip from GitHub release and unzip
run: |
wget https://github.com/mikel-brostrom/boxmot/releases/download/v10.0.83/runs.zip -O run.zip
unzip run.zip -d run
- name: Evaluation and Summarize Results
run: |
if python3 tracking/val.py --benchmark MOT17-mini --yolo-model yolov8n.pt --reid-model osnet_x0_25_msmt17.pt --tracking-method ${{ matrix.tracker }} --verbose --source ./assets/MOT17-mini/train --ci; then
if python3 tracking/val.py --benchmark MOT17-mini --yolo-model yolov8x.pt --reid-model osnet_x1_0_dukemtmcreid.pt --tracking-method ${{ matrix.tracker }} --verbose --source ./assets/MOT17-mini/train --ci; then
STATUS="✅"
else
STATUS="❌"
Expand Down

0 comments on commit 115e207

Please sign in to comment.