Commit 2c34611d authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

bugfixed: cvpr2019 ==> detection_cvpr_2019

parent c1d93158
...@@ -110,7 +110,7 @@ class NuScenesDataset(DatasetTemplate): ...@@ -110,7 +110,7 @@ class NuScenesDataset(DatasetTemplate):
def get_template_prediction(num_samples): def get_template_prediction(num_samples):
ret_dict = { ret_dict = {
'name': np.zeros(num_samples), 'score': np.zeros(num_samples), 'name': np.zeros(num_samples), 'score': np.zeros(num_samples),
'boxes_lidar': np.zeros([num_samples, 7]) 'boxes_lidar': np.zeros([num_samples, 7]), 'pred_labels': np.zeros(num_samples)
} }
return ret_dict return ret_dict
...@@ -166,7 +166,7 @@ class NuScenesDataset(DatasetTemplate): ...@@ -166,7 +166,7 @@ class NuScenesDataset(DatasetTemplate):
from nuscenes.eval.detection.config import config_factory from nuscenes.eval.detection.config import config_factory
from nuscenes.eval.detection.evaluate import NuScenesEval from nuscenes.eval.detection.evaluate import NuScenesEval
eval_version = 'cvpr_2019' eval_version = 'detection_cvpr_2019'
eval_set_map = { eval_set_map = {
'v1.0-mini': 'mini_val', 'v1.0-mini': 'mini_val',
'v1.0-trainval': 'val', 'v1.0-trainval': 'val',
......
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