Skip to content

GitLab

Unverified Commit 19068b52 authored by ArtificialCleverness's avatar ArtificialCleverness Committed by GitHub
Browse files

modified few documentation about ious (#550) 


Co-authored-by: default avatarpengfengming <pengfengming@haomo.ai>
parent 7efdf026
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
pcdet/ops/iou3d_nms/iou3d_nms_utils.py
View file @ 19068b52
......@@ -13,10 +13,10 @@ def boxes_bev_iou_cpu(boxes_a, boxes_b):
"""
Args:
boxes_a: (N, 7) [x, y, z, dx, dy, dz, heading]
boxes_b: (N, 7) [x, y, z, dx, dy, dz, heading]
boxes_b: (M, 7) [x, y, z, dx, dy, dz, heading]
Returns:
ans_iou: (N, M)
"""
boxes_a, is_numpy = common_utils.check_numpy_to_torch(boxes_a)
boxes_b, is_numpy = common_utils.check_numpy_to_torch(boxes_b)
......@@ -32,7 +32,7 @@ def boxes_iou_bev(boxes_a, boxes_b):
"""
Args:
boxes_a: (N, 7) [x, y, z, dx, dy, dz, heading]
boxes_b: (N, 7) [x, y, z, dx, dy, dz, heading]
boxes_b: (M, 7) [x, y, z, dx, dy, dz, heading]
Returns:
ans_iou: (N, M)
......@@ -49,7 +49,7 @@ def boxes_iou3d_gpu(boxes_a, boxes_b):
"""
Args:
boxes_a: (N, 7) [x, y, z, dx, dy, dz, heading]
boxes_b: (N, 7) [x, y, z, dx, dy, dz, heading]
boxes_b: (M, 7) [x, y, z, dx, dy, dz, heading]
Returns:
ans_iou: (N, M)
......
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