BoxMOT tracking backend and YOLO training module - #22
Conversation
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
Wire BoxMOT in as the tracking backend and move YOLO dataset/training out of data_utils into a dedicated src/yolo/ module, exposed via create_yolo_dataset and train_yolo_model. Replace the polynomial-regression bbox dimension model with a grid-based template using linear interpolation, which is faster and behaves better outside the fitted range. Also ignore the working videos, root-level scratch images, output/ and the vendored boxmot/ clone, none of which belong in history. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
|
Rebased onto current
# Moved out of data_utils into a dedicated src/yolo module.
"create_yolo_dataset": "src.yolo:create_yolo_dataset",
"train_yolo_model": "src.yolo:train_yolo_model",Verified:
Unchanged from the original: this is your pre-existing working-tree code committed as found. Not reviewed for correctness and not executed. The |
24d4860 to
c89a4a4
Compare
Atom's uncommitted working-tree changes from before this session, committed and pushed so they are reviewable rather than at risk on a single machine.
What it contains
boxmot>=11.0.8added topyproject, plusscripts/run_tracking.sh.src/yolo/module —dataset.pyandtrain.py, exposed ascreate_yolo_datasetandtrain_yolo_model, moving YOLO dataset/training out ofdata_utils. Plusscripts/prepare_and_train_yolo.sh.analyze_bbox_dimensions.pyrewritten — the polynomial-regression bbox-size model replaced with a grid-based template using linear interpolation, which is faster and behaves better outside the fitted range.scripts/plot_pitch_coordinates.py— new..gitignorehardening — root-level working videos,output/, root scratch images, and the vendoredboxmot/clone. Before this, ~220 MB of loose.mp4files in the repo root were untracked and unignored, so anygit add -Awould have committed them into history permanently.Status
Committed as found — not reviewed for correctness and not run. The
.gitignorepart is the piece worth landing regardless, since it removes a standing footgun.Note this branch predates the match-ID correction, so if it touches any hardcoded match ID it may need a rebase after #20 and #21 land.
🤖 Generated with Claude Code