Indexes out of range

[Manthata]

Hello, would someone kindly help me with this error

DeprecationWarning: The linear_assignment_ module is deprecated in 0.21 and will be removed from 0.23. Use scipy.optimize.linear_sum_assignment instead.
DeprecationWarning)
Video size 1280 720
2
Traceback (most recent call last):
File "object_tracker.py", line 105, in
cls = classes[int(cls_pred)]
IndexError: list index out of range

[dr-moo]

Second this! I am getting the same issue for both single image detection and video tracking

@cfotache, is there any chance you could elaborate a bit because the whole idea of using custom trained YOLO with your implementation (the best one I found at least) of SORT is very tempting...

[Manthata]

Hey! I think you should try to change the modules from cuuda to normal Float.Tensor and some part of the program so that it is runs on a local machine without GPU settings. If you are using a GPU then it should work just fun without making any changes

[Manthata]

I just recloned again and changed the settings from GPU to CPU and it worked fine on my local computer.

[dr-moo]

I just recloned again and changed the settings from GPU to CPU and it worked fine on my local computer.

The problem is that i do have a GPU with cuda compiled and working on other projects, but i still get the same out of index error when i try to run object_tracker.py

Which parts of the code are responsible for CPU-only runs?

Thanks in advance!

[archimanlol]

The linear_assignment function is deprecated in 0.21 and will be removed from 0.23, but sklearn.utils.linear_assignment_ can be replaced by scipy.optimize.linear_sum_assignment You can use from scipy.optimize import linear_sum_assignment as linear_assignment.