Commit b107238d authored by zhangwenwei's avatar zhangwenwei
Browse files

Merge branch 'clean-iou' into 'master'

clean iou calculation

See merge request open-mmlab/mmdet.3d!84
parents b2c43ffd 8c4c9aee
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(): ...@@ -425,7 +425,10 @@ def test_multi_class_nms():
1.4715e+00, -2.9639e+00 1.4715e+00, -2.9639e+00
]]).cuda() ]]).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() expected_selected = torch.Tensor([0, 1, 4, 8]).cuda()
assert torch.all(selected == expected_selected) 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