Error in running with flag_nms= True

[Atishaysjain]

When I run demo_video_mobile.py after setting flag_nms = True I get the following error :-
TypeError: list indices must be integers or slices, not tuple

For the following code inside the function def apply_nms :-
keep = np.where((cls_dets[:, 4] >= min_scores) &
((cls_dets[:, 3] - cls_dets[:, 1]) * (cls_dets[:, 2] - cls_dets[:, 0]) >= min_box_size))[0]

When I first convert cls_dets into numpy array by cls_dets = np.array(cls_dets) and then execute, the following error is shown :-
IndexError: too many indices for array

This is because the shape of cls_dets is (4,) while the code written is for a 2 dimensional array.

Please suggest a solution so that I can apply non max suppresion with this code.

[JosePeeterson]

Hi Atishaysjain,
could you please share your env dependencies file with all the version numbers? and the cuda toolkit version.

When you first ran demo_video_mobile.py did you see the following error?

Traceback (most recent call last): File "demo_video_mobile.py", line 28, in <module> from nms.gpu_nms import gpu_nms ImportError: /home/peeterson/Desktop/lighttrack/HPE/../lib/nms/gpu_nms.cpython-36m-x86_64-linux-gnu.so: undefined symbol: __cudaPopCallConfiguration

If you did could you share how you solved it? thanks.