Unverified Commit 9b091154 authored by yinchimaoliang's avatar yinchimaoliang Committed by GitHub
Browse files

Move use_valid_flag to nus (#106)

parent 076b2cbd
...@@ -142,10 +142,7 @@ def create_groundtruth_database(dataset_class_name, ...@@ -142,10 +142,7 @@ def create_groundtruth_database(dataset_class_name,
""" """
print(f'Create GT Database of {dataset_class_name}') print(f'Create GT Database of {dataset_class_name}')
dataset_cfg = dict( dataset_cfg = dict(
type=dataset_class_name, type=dataset_class_name, data_root=data_path, ann_file=info_path)
data_root=data_path,
ann_file=info_path,
use_valid_flag=True)
if dataset_class_name == 'KittiDataset': if dataset_class_name == 'KittiDataset':
file_client_args = dict(backend='disk') file_client_args = dict(backend='disk')
dataset_cfg.update( dataset_cfg.update(
...@@ -171,19 +168,21 @@ def create_groundtruth_database(dataset_class_name, ...@@ -171,19 +168,21 @@ def create_groundtruth_database(dataset_class_name,
]) ])
elif dataset_class_name == 'NuScenesDataset': elif dataset_class_name == 'NuScenesDataset':
dataset_cfg.update(pipeline=[ dataset_cfg.update(
dict(type='LoadPointsFromFile', load_dim=5, use_dim=5), use_valid_flag=True,
dict( pipeline=[
type='LoadPointsFromMultiSweeps', dict(type='LoadPointsFromFile', load_dim=5, use_dim=5),
sweeps_num=10, dict(
use_dim=[0, 1, 2, 3, 4], type='LoadPointsFromMultiSweeps',
pad_empty_sweeps=True, sweeps_num=10,
remove_close=True), use_dim=[0, 1, 2, 3, 4],
dict( pad_empty_sweeps=True,
type='LoadAnnotations3D', remove_close=True),
with_bbox_3d=True, dict(
with_label_3d=True) type='LoadAnnotations3D',
]) with_bbox_3d=True,
with_label_3d=True)
])
dataset = build_dataset(dataset_cfg) dataset = build_dataset(dataset_cfg)
if database_save_path is None: if database_save_path is None:
......
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