Skip to content

GitLab

Unverified Commit e3327ca6 authored by q.yao's avatar q.yao Committed by GitHub
Browse files

Fix iou3d python bind (#1524)

parent 58e32423
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
mmcv/ops/csrc/pytorch/pybind.cpp
View file @ e3327ca6
...@@ -119,10 +119,10 @@ void iou3d_boxes_overlap_bev_forward(Tensor boxes_a, Tensor boxes_b, ...@@ -119,10 +119,10 @@ void iou3d_boxes_overlap_bev_forward(Tensor boxes_a, Tensor boxes_b,
void iou3d_boxes_iou_bev_forward(Tensor boxes_a, Tensor boxes_b, void iou3d_boxes_iou_bev_forward(Tensor boxes_a, Tensor boxes_b,
Tensor ans_iou); Tensor ans_iou);
int iou3d_nms_forward(Tensor boxes, Tensor keep, Tensor keep_num, void iou3d_nms_forward(Tensor boxes, Tensor keep, Tensor keep_num,
float nms_overlap_thresh); float nms_overlap_thresh);
int iou3d_nms_normal_forward(Tensor boxes, Tensor keep, Tensor keep_num, void iou3d_nms_normal_forward(Tensor boxes, Tensor keep, Tensor keep_num,
float nms_overlap_thresh); float nms_overlap_thresh);
void furthest_point_sampling_forward(Tensor points_tensor, Tensor temp_tensor, void furthest_point_sampling_forward(Tensor points_tensor, Tensor temp_tensor,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment