Skip to content

Commit

Permalink
Merge pull request #1651 from mikel-brostrom/conf-fix
Browse files Browse the repository at this point in the history
original configs
  • Loading branch information
mikel-brostrom authored Sep 20, 2024
2 parents 4fdbaad + 89d8fd5 commit 4610640
Show file tree
Hide file tree
Showing 8 changed files with 26 additions and 29 deletions.
12 changes: 9 additions & 3 deletions .github/workflows/benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ jobs:
# Specify the path where the dataset is stored
path: tracking/val_utils/MOT17-50.zip
# Create a cache key, you can use a fixed key if the dataset is static
key: mot17-50-dataset-cache
key: mot17-50-dataset-cache-v3

- name: Download eval data
run: |
Expand All @@ -79,11 +79,15 @@ jobs:
uses: actions/cache@v3
with:
path: tracking/val_utils/MOT17-50.zip
key: mot17-50-dataset-cache
key: mot17-50-dataset-cache-v3

- name: Evaluation and Summarize Results
run: |
pwd
ls tracking/val_utils
ls tracking/val_utils/data
if python3 tracking/val.py --imgsz 320 --classes 0 --benchmark MOT17-50 --yolo-model yolov8x.pt --reid-model osnet_x1_0_dukemtmcreid.pt --tracking-method ${{ matrix.tracker }} --verbose --source ./tracking/val_utils/data/MOT17-50/train; then
STATUS="✅"
else
Expand Down Expand Up @@ -204,14 +208,16 @@ jobs:
git add README.md
git commit -m "Overwrote tracker results in README.md"
git push origin "$BRANCH_NAME"
env:
BRANCH_NAME: ${{ steps.create_branch.outputs.BRANCH_NAME }}

# Create a pull request (PR)
- name: Create Pull Request
if: steps.check_changes.outputs.changed == 'true'
uses: peter-evans/create-pull-request@v5
with:
token: ${{ secrets.GITHUB_TOKEN }}
branch: ${{ github.ref }}
branch: ${{ env.BRANCH_NAME }} # Use the created branch
commit-message: Overwrote tracker results in README.md
title: Overwrite tracker results in README.md
body: "This PR updates the tracker results table in the README.md."
19 changes: 8 additions & 11 deletions boxmot/configs/botsort.yaml
Original file line number Diff line number Diff line change
@@ -1,13 +1,10 @@
# Trial number: 0
# HOTA, MOTA, IDF1: [37.804, 19.34, 34.483]
appearance_thresh: 0.7636154550757781
cmc_method: sparseOptFlow
conf: 0.4828271520496597
appearance_thresh: 0.25
cmc_method: ecc
frame_rate: 30
lambda_: 0.9764964093204093
match_thresh: 0.22182609980772128
new_track_thresh: 0.25855770885476126
proximity_thresh: 0.37262205760353817
track_buffer: 40
track_high_thresh: 0.5
lambda_: 0.98
match_thresh: 0.8
new_track_thresh: 0.7
proximity_thresh: 0.5
track_buffer: 30
track_high_thresh: 0.6
track_low_thresh: 0.1
1 change: 0 additions & 1 deletion boxmot/configs/bytetrack.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,3 @@ track_thresh: 0.5 # tracking confidence threshold, should be the same as model
track_buffer: 30 # the frames for keep lost tracks
match_thresh: 0.8 # matching threshold for tracking
frame_rate: 30 # FPS
conf_thres: 0.5122620708221085
9 changes: 3 additions & 6 deletions boxmot/configs/deepocsort.yaml
Original file line number Diff line number Diff line change
@@ -1,18 +1,15 @@
# Trial number: 0
# HOTA, MOTA, IDF1: [27.884, 11.425, 21.545]
alpha_fixed_emb: 0.95
asso_func: giou
asso_func: iou
aw_off: false
aw_param: 0.5
cmc_off: false
conf: 0.5
delta_t: 3
det_thresh: 0
det_thresh: 0.5
embedding_off: false
inertia: 0.2
iou_thresh: 0.3
max_age: 30
min_hits: 1
min_hits: 3
w_association_emb: 0.75
Q_xy_scaling: 0.01
Q_s_scaling: 0.0001
2 changes: 1 addition & 1 deletion boxmot/configs/imprassoc.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,4 +9,4 @@ proximity_thresh: 0.1
second_match_thresh: 0.19
track_buffer: 35
track_high_thresh: 0.5
track_low_thresh: 0.1
track_low_thresh: 0.1
9 changes: 3 additions & 6 deletions boxmot/configs/ocsort.yaml
Original file line number Diff line number Diff line change
@@ -1,13 +1,10 @@
# Trial number: 137
# HOTA, MOTA, IDF1: [55.567]
asso_func: giou
conf_thres: 0.5122620708221085
asso_func: iou
delta_t: 3
det_thresh: 0
det_thresh: 0.6
inertia: 0.2
iou_thresh: 0.3
max_age: 30
min_hits: 1
min_hits: 3
use_byte: false
Q_xy_scaling: 0.01
Q_s_scaling: 0.0001
2 changes: 1 addition & 1 deletion boxmot/configs/strongsort.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,5 @@ max_age: 30
max_cos_dist: 0.3
max_iou_dist: 0.7
mc_lambda: 0.98
n_init: 1
n_init: 3
nn_budget: 100
1 change: 1 addition & 0 deletions tracking/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import numpy as np
import torch
import time
import pandas as pd
from ultralytics.utils import ops
from ultralytics.engine.results import Results
Expand Down

0 comments on commit 4610640

Please sign in to comment.