From 9b09115457adfa02c6b80a92b3250f9438ad961c Mon Sep 17 00:00:00 2001 From: yinchimaoliang Date: Wed, 16 Sep 2020 10:05:03 +0800 Subject: [PATCH] Move use_valid_flag to nus (#106) --- tools/data_converter/create_gt_database.py | 33 +++++++++++----------- 1 file changed, 16 insertions(+), 17 deletions(-) diff --git a/tools/data_converter/create_gt_database.py b/tools/data_converter/create_gt_database.py index d0326417d..2150d20c0 100644 --- a/tools/data_converter/create_gt_database.py +++ b/tools/data_converter/create_gt_database.py @@ -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: