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