AttributeError: 'DeepOCSort' object has no attribute 'reset'
petem24 opened this issue · 1 comments
Search before asking
- I have searched the Yolo Tracking issues and discussions and found no similar questions.
Yolo Tracking Component
Tracking
Bug
python tracking/track.py --source crash.mp4
Downloading https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt to '/home/peter/Documents/repos/boxmot/tracking/weights/yolov8n.pt'...
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 6.25M/6.25M [00:00<00:00, 23.2MB/s]
2024-08-27 15:42:31.952 | INFO | boxmot.utils.torch_utils:select_device:52 - Yolo Tracking v10.0.78 🚀 Python-3.11.9 torch-2.2.2+cu121
CUDA:0 (NVIDIA GeForce RTX 3060 Ti, 7972MiB)
2024-08-27 15:42:31.989 | SUCCESS | boxmot.appearance.reid_model_factory:load_pretrained_weights:183 - Loaded pretrained weights from /home/peter/Documents/repos/boxmot/tracking/weights/osnet_x0_25_msmt17.pt
Traceback (most recent call last):
File "/home/peter/Documents/repos/boxmot/tracking/track.py", line 172, in
run(opt)
File "/home/peter/Documents/repos/boxmot/.venv/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/home/peter/Documents/repos/boxmot/tracking/track.py", line 101, in run
for r in results:
File "/home/peter/Documents/repos/boxmot/.venv/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 35, in generator_context
response = gen.send(None)
^^^^^^^^^^^^^^
File "/home/peter/Documents/repos/boxmot/.venv/lib/python3.11/site-packages/ultralytics/engine/predictor.py", line 265, in stream_inference
self.run_callbacks("on_predict_postprocess_end")
File "/home/peter/Documents/repos/boxmot/.venv/lib/python3.11/site-packages/ultralytics/engine/predictor.py", line 402, in run_callbacks
callback(self)
File "/home/peter/Documents/repos/boxmot/.venv/lib/python3.11/site-packages/ultralytics/trackers/track.py", line 74, in on_predict_postprocess_end
tracker.reset()
^^^^^^^^^^^^^
AttributeError: 'DeepOCSort' object has no attribute 'reset'
Environment
Commit 7d79be7
Python 3.11.9
All installed inside of venv
Minimal Reproducible Example
poetry install --with yolo
poetry shell
pip install boxmot
python tracking/track.py --source crash.mp4
This shouldn't be an issue when installing as specified under README 😄. Notice you have to install the custom Ultralytics package that this repo is pointing to