Commit 7a43a52a authored by liyinhao's avatar liyinhao
Browse files

change names

parent f7cf2bb0
from .class_names import dataset_aliases, get_classes, kitti_classes
from .indoor_utils import indoor_eval
from .kitti_utils import kitti_eval, kitti_eval_coco_style
__all__ = [
'dataset_aliases', 'get_classes', 'kitti_classes', 'kitti_eval_coco_style',
'kitti_eval'
'kitti_eval', 'indoor_eval'
]
from .eval import indoor_eval
__all__ = ['indoor_eval']
......@@ -365,7 +365,7 @@ def boxes3d_depth_to_lidar(boxes3d, mid_to_bottom=True):
return boxes3d_lidar
def scannet_eval(gt_annos, dt_annos, metric, class2type):
def indoor_eval(gt_annos, dt_annos, metric, class2type):
"""Scannet Evaluation.
Evaluate the result of the detection.
......
......@@ -211,13 +211,13 @@ class ScannetDataset(torch_data.Dataset):
def evaluate(self, results, metric=None, logger=None, pklfile_prefix=None):
results = self._format_results(results)
from mmdet3d.core.evaluation.scannet_utils.eval import scannet_eval
from mmdet3d.core.evaluation import indoor_eval
assert ('AP_IOU_THRESHHOLDS' in metric)
gt_annos = [
copy.deepcopy(info['annos']) for info in self.scannet_infos
]
ap_result_str, ap_dict = scannet_eval(gt_annos, results, metric,
self.class2type)
ap_result_str, ap_dict = indoor_eval(gt_annos, results, metric,
self.class2type)
return ap_dict
def __len__(self):
......
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