Skip to content

GitLab

Commit c45634a3 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

bugfixed: try two eval_version for different version of nuscenes-devkit

parent 9c07c58a
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 2 deletions
pcdet/datasets/nuscenes/nuscenes_dataset.py
View file @ c45634a3
...@@ -210,16 +210,21 @@ class NuScenesDataset(DatasetTemplate): ...@@ -210,16 +210,21 @@ 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 = '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',
'v1.0-test': 'test' 'v1.0-test': 'test'
} }
try:
eval_version = 'detection_cvpr_2019'
eval_config = config_factory(eval_version)
except:
eval_version = 'cvpr_2019'
eval_config = config_factory(eval_version)
nusc_eval = NuScenesEval( nusc_eval = NuScenesEval(
nusc, nusc,
config=config_factory(eval_version), config=eval_config,
result_path=res_path, result_path=res_path,
eval_set=eval_set_map[self.dataset_cfg.VERSION], eval_set=eval_set_map[self.dataset_cfg.VERSION],
output_dir=str(output_path), output_dir=str(output_path),
......
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