changes + track

This commit is contained in:
√(noham)²
2024-07-23 16:09:32 +02:00
parent db075fb427
commit 25207230c6
124 changed files with 7850 additions and 26104 deletions

36
yolov7-setup/track.md Normal file
View File

@@ -0,0 +1,36 @@
Images :
```bash
python3 tracker/track_demo.py --device cpu --obj /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/tracking_images --detector yolov7 --tracker sort --detector_model_path /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/yolov7x-datasetv2_2024072211/weights/best.pt --save_dir /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/tracking --save_images
```
Video :
```bash
python3 tracker/track_demo.py --device cpu --obj /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/sources/2016-100m.mp4 --detector yolov7 --tracker sort --detector_model_path /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/yolov7x-datasetv2_2024072211/weights/best.pt --save_dir /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/tracking --save_videos
```
```bash
python3 tracker/track_demo.py --device cpu --obj /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/vids/400m_2.mp4 --detector yolov7 --tracker sort --detector_model_path /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/yolov7x-datasetv2_2024072211/weights/best.pt --save_dir /Users/noham/Documents/GitHub/Stage-2024/VoTT-v2/tracking --save_videos
```
Fixes :
self.other_param['stride'] -> 32
np.float64 in /tracker/trackers/tracklet.py
File "/Users/noham/Documents/GitHub/Stage-2024/test/yolov7-tracker/tracker/trackers/tracklet.py", line 44, in __init__
self.kalman_filter = MOTION_MODEL_DICT[motion]()
KeyError: 'default'
line 43 :
self.motion = 'sort'
motion = 'sort'
```python
TRACKER_DICT = {
'sort': SortTracker,
'bytetrack': ByteTracker,
'botsort': BotTracker,
'c_bioutrack': C_BIoUTracker,
'ocsort': OCSortTracker,
'deepsort': DeepSortTracker
}
```