OBBDetection
OBBDetection copied to clipboard
OBBDetection/mmdet/ops/nms_rotated/nms_rotated_wrapper.py", line 41, in obb_nms ori_inds = ori_inds[~too_small]
in /home/ym/OBBDetection/mmdet/ops/nms_rotated/nms_rotated_wrapper.py in Line40 应该添加: ori_inds = ori_inds.to(too_small.device) 否则会报设备不一致的错误