nms_rotated npu add new attr is_angle

open-mmlab · Jul 16, 2024 · 47dff6f · 47dff6f
1 parent cf8ffc2
commit 47dff6f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/mmcv/ops/csrc/pytorch/npu/nms_rotated_npu.cpp b/mmcv/ops/csrc/pytorch/npu/nms_rotated_npu.cpp
@@ -26,6 +26,7 @@ Tensor nms_rotated_npu(const Tensor dets, const Tensor scores,
       .Output(selectedBox)
       .Output(selectedIndex)
       .Attr("iou_threshold", (float)iou_threshold)
+      .Attr("is_angle", false)
       .Run();
   selectedIndex = selectedIndex.to(at::kLong);
   return selectedIndex;

diff --git a/mmcv/ops/nms.py b/mmcv/ops/nms.py
@@ -460,10 +460,6 @@ def nms_rotated(dets: Tensor,
         input_labels = labels
     if dets.device.type in ('npu', 'mlu'):
         order = scores.new_empty(0, dtype=torch.long)
-        if dets.device.type == 'npu':
-            coefficient = 57.29578  # 180 / PI
-            for i in range(dets.size()[0]):
-                dets_cw[i][4] *= coefficient  # radians to angle
         keep_inds = ext_module.nms_rotated(dets_cw, scores, order, dets_cw,
                                            input_labels, iou_threshold,
                                            multi_label)