AttributeError when using Strong Sort tracker in TorchYOLO

[userkw]

When attempting to use the Strong Sort tracker with the TorchYOLO framework, users encounter an AttributeError. The error occurs when executing the following command:

torchyolo tracker --config_path torchyolo/configs/default_config.yaml --model_type yolov5 --model_path yolov5s.pt --tracker_config_path torchyolo/configs/tracker/strong_sort.yaml --source yl.mp4
Error Message:
YOLOv5 2024-3-25 Python-3.8.19 torch-1.13.0+cpu CPU

Fusing layers...
Model summary: 214 layers, 7022326 parameters, 0 gradients, 15.9 GFLOPs
Adding AutoShape...
Traceback (most recent call last):
File "C:\Users\elkis\anaconda3\envs\env2\lib\runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "C:\Users\elkis\anaconda3\envs\env2\lib\runpy.py", line 87, in run_code
exec(code, run_globals)
File "C:\Users\elkis\anaconda3\envs\env2\Scripts\torchyolo.exe_main.py", line 7, in
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\cli.py", line 10, in app
Fire(torchyolo_app)
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\fire\core.py", line 141, in Fire
component_trace = _Fire(component, args, parsed_flag_args, context, name)
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\fire\core.py", line 475, in _Fire
component, remaining_args = _CallAndUpdateTrace(
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\fire\core.py", line 691, in _CallAndUpdateTrace
component = fn(*varargs, **kwargs)
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\predict.py", line 88, in tracker_predict
result = model.tracker_predict(
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\predict.py", line 56, in tracker_predict
tracker_prediction = self.model.tracker_predict(
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\modelhub\yolov5.py", line 78, in tracker_predict
tracker_module = load_tracker_module(
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\tracker\tracker_module.py", line 18, in load_tracker_module
tracker_module = load_tracker(
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\tracker\tracker_zoo.py", line 148, in load_tracker
tracker_module = create_tracker(
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\tracker\tracker_zoo.py", line 79, in create_tracker
distance_function=config.NORFAIR_TRACK.DISTANCE_FUNCTION,
AttributeError: 'YamlParser' object has no attribute 'NORFAIR_TRACK'

[kadirnar]

Can you share your code?

[userkw]

@kadirnar I'm just using this command line: :
torchyolo tracker --config_path torchyolo/configs/default_config.yaml
--model_type yolov5 --model_path yolov5s.pt
--tracker_config_path norfair.yaml
I'm using my custom YOLOv5 model specified here: --model_path yolov5s.pt, which is originally best.pt, but I renamed it to yolov5s.
The problem arises when I use --tracker_config_path norfair.yaml with a strong_sort or any other tracker. I encounter this error:
tracker_module = create_tracker(
File "C:\Users\elkis\anaconda3\envs\env2\lib\site-packages\torchyolo\tracker\tracker_zoo.py", line 79, in create_tracker
distance_function=config.NORFAIR_TRACK.DISTANCE_FUNCTION,
AttributeError: 'YamlParser' object has no attribute 'NORFAIR_TRACK'