search

garrickbrazil / M3D-RPN

MIT License
261 stars 67 forks source link

The nms problem #52

Open Nicoladao opened 2 years ago

Nicoladao commented 2 years ago

I test the code,it seems that nms does not work.The effect is shown in the figure below： and can not detect object in far distance.Is anybody know how to fix it? Thank you! image

qaazii commented 2 years ago

You need to input correct calibration matrix files

KenYu910645 commented 2 years ago

I had a similar problem and solved it by changing gpu_nms() to py_cpu_nms(). For anyone who want to try this method:

  1. Edit rpn_util.py at Line 16 # from lib.nms.gpu_nms import gpu_nms from lib.nms.py_cpu_nms import py_cpu_nms
  2. Edit rpn_util.py at Line 1281 # keep_inds = gpu_nms(aboxes[:, 0:5].astype(np.float32), rpn_conf.nms_thres, device_id=gpu) keep_inds = py_cpu_nms(aboxes[:, 0:5].astype(np.float32), rpn_conf.nms_thres)

My guess is this issue is caused by Cython which can't get correct version of CUDA to compile with. But I'm not particularly familiar with CUDA nor Cython, hence, I choose to workaround this problem.

Rishav-Paramhans commented 1 year ago

I also has a similar issue and it wasn't resolved even after using py_cpu_nms.