Unverified Commit 23eb359b authored by gy77's avatar gy77 Committed by GitHub
Browse files

Add type hint for the "boxes_iou_bev" function in iou3d.py (#2005)

* add type hints in iou3d.py

* fix lint

* Add a missing item in iou3d.py
parent d0dc7577
...@@ -100,7 +100,7 @@ def _xyxyr2xywhr(boxes: Tensor) -> Tensor: ...@@ -100,7 +100,7 @@ def _xyxyr2xywhr(boxes: Tensor) -> Tensor:
dim=-1) dim=-1)
def boxes_iou_bev(boxes_a, boxes_b): def boxes_iou_bev(boxes_a: Tensor, boxes_b: Tensor) -> Tensor:
"""Calculate boxes IoU in the Bird's Eye View. """Calculate boxes IoU in the Bird's Eye View.
Args: Args:
......
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