Updated tracking to support multi-lemon tracking

- Improved tracking logic to handle multiple lemons simultaneously
- Added necessary YAML configuration for BoT-SORT

Higher Sensitivity to Minor Defects:
- Weighted “DefectiveLemon” more heavily
- Extended HISTORY_LENGTH for improved tracking stability
This commit is contained in:
charlotte
2025-03-12 14:37:48 -04:00
parent ffb7a6300a
commit 061d049bdf
2 changed files with 146 additions and 0 deletions

21
botsort.yaml Normal file
View File

@@ -0,0 +1,21 @@
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# Default Ultralytics settings for BoT-SORT tracker when using mode="track"
# For documentation and examples see https://docs.ultralytics.com/modes/track/
# For BoT-SORT source code see https://github.com/NirAharon/BoT-SORT
tracker_type: botsort # tracker type, ['botsort', 'bytetrack']
track_high_thresh: 0.25 # threshold for the first association
track_low_thresh: 0.1 # threshold for the second association
new_track_thresh: 0.4 # threshold for init new track if the detection does not match any tracks
track_buffer: 30 # buffer to calculate the time when to remove tracks
match_thresh: 0.7 # threshold for matching tracks
fuse_score: True # Whether to fuse confidence scores with the iou distances before matching
# min_box_area: 10 # threshold for min box areas(for tracker evaluation, not used for now)
# BoT-SORT settings
gmc_method: sparseOptFlow # method of global motion compensation
# ReID model related thresh (not supported yet)
proximity_thresh: 0.5
appearance_thresh: 0.25
with_reid: False