Commit 8c4c9aee authored by zhangwenwei's avatar zhangwenwei
Browse files

clean iou calculation

parent b2c43ffd
import logging
from mmcv.utils import get_logger
def get_root_logger(log_file=None, log_level=logging.INFO):
logger = get_logger(name='mmdet3d', log_file=log_file, log_level=log_level)
return logger
......@@ -425,7 +425,10 @@ def test_multi_class_nms():
1.4715e+00, -2.9639e+00
]]).cuda()
selected = self.multi_class_nms(box_probs, box_preds, 0.1, 0.001)
input_meta = dict(
box_type_3d=LiDARInstance3DBoxes, box_mode_3d=Box3DMode.LIDAR)
selected = self.multi_class_nms(box_probs, box_preds, 0.1, 0.001,
input_meta)
expected_selected = torch.Tensor([0, 1, 4, 8]).cuda()
assert torch.all(selected == expected_selected)
......
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