OBBDetection icon indicating copy to clipboard operation
OBBDetection copied to clipboard

OBBDetection/mmdet/ops/nms_rotated/nms_rotated_wrapper.py", line 41, in obb_nms ori_inds = ori_inds[~too_small]

Open binfenhuayun opened this issue 4 months ago • 0 comments

in /home/ym/OBBDetection/mmdet/ops/nms_rotated/nms_rotated_wrapper.py in Line40 应该添加: ori_inds = ori_inds.to(too_small.device) 否则会报设备不一致的错误

binfenhuayun avatar Oct 15 '24 03:10 binfenhuayun