Open kristupas-g opened 7 months ago
Did you fix this problem? I met the same problem
I have solve it! Just find the file "nms_rotated.py" and replace the code in line 44
box_inds = ext_module.nms_rotated(_boxes, _scores, order, dets_sorted, iou_threshold, 0)
with
labels = torch.ones_like(_scores) * cls_id box_inds = ext_module.nms_rotated(_boxes, _scores, order, dets_sorted, labels, iou_threshold, 0)
Checklist
Describe the bug
I am trying to convert a Oriented RCNN model to ONNX runtime. I am running the pretty much the same as script as in the documentation. What am I missing? The input image was downloaded off the MMRotate library and the checkpoint was downloaded using
mim
.Reproduction
In google colab enviroment running this code:
Environment