Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

download dets n embs from release #1639

Merged
merged 4 commits into from
Sep 18, 2024
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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
Loading