Skip to content

GitLab

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

Move use_valid_flag to nus (#106)

parent 076b2cbd
Show whitespace changes
Inline Side-by-side
Showing with 16 additions and 17 deletions
tools/data_converter/create_gt_database.py
View file @ 9b091154
...@@ -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,7 +168,9 @@ def create_groundtruth_database(dataset_class_name, ...@@ -171,7 +168,9 @@ def create_groundtruth_database(dataset_class_name,
]) ])
elif dataset_class_name == 'NuScenesDataset': elif dataset_class_name == 'NuScenesDataset':
dataset_cfg.update(pipeline=[ dataset_cfg.update(
use_valid_flag=True,
pipeline=[
dict(type='LoadPointsFromFile', load_dim=5, use_dim=5), dict(type='LoadPointsFromFile', load_dim=5, use_dim=5),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
......
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