Merge pull request #70 from ZrBac/patch-2

update points_in_boxes_all
open-mmlab · Nov 6, 2024 · e3cf445 · e3cf445
2 parents 96c4d7c + 6c48402
commit e3cf445
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/mmcv/ops/points_in_boxes.py b/mmcv/ops/points_in_boxes.py
@@ -133,8 +133,6 @@ def points_in_boxes_all(points: Tensor, boxes: Tensor) -> Tensor:
     if points.device.type != 'npu':
         if torch.cuda.current_device() != points_device:
             torch.cuda.set_device(points_device)
-    elif points.device.type == 'npu':
-        boxes[:, :, 2] += boxes[:, :, 5] / 2.0
     ext_module.points_in_boxes_all_forward(boxes.contiguous(),
                                            points.contiguous(),
                                            box_idxs_of_pts)