M3D-RPN
M3D-RPN copied to clipboard
The nms problem
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!
You need to input correct calibration matrix files
I had a similar problem and solved it by changing gpu_nms() to py_cpu_nms(). For anyone who want to try this method:
- 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
- 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.
I also has a similar issue and it wasn't resolved even after using py_cpu_nms.