No module named 'mmdet3d.ops.iou3d' about uvtr HOT 2 CLOSED

Hi, I guess you are using the mmdet3d with version >0.17.3 (required) here. A simple solution is to change the code from mmdet3d.ops.iou3d.iou3d_utils import nms_gpu, nms_normal_gpu to from mmdet3d.ops import nms and also modify the corresponding function nms_gpu to nms

from uvtr.

I think the replaced code could be

    if test_cfg.use_rotate_nms:
        nms_func = nms_bev
    else:
        nms_func = nms_normal_bev

and the import directives can use from mmdet3d.core.post_processing import nms_bev, nms_normal_bev or from mmcv.ops import nms_bev, nms_normal_bev

from uvtr.