From 11fb55d43f03a7f3da8855f182eded9ac118bbbf Mon Sep 17 00:00:00 2001 From: inter Date: Sun, 21 Sep 2025 20:19:50 +0800 Subject: [PATCH] Add File --- .../cfgs/dataset_configs/custom_dataset.yaml | 70 +++++++++++++++++++ 1 file changed, 70 insertions(+) create mode 100644 tools/cfgs/dataset_configs/custom_dataset.yaml diff --git a/tools/cfgs/dataset_configs/custom_dataset.yaml b/tools/cfgs/dataset_configs/custom_dataset.yaml new file mode 100644 index 0000000..f8fedc9 --- /dev/null +++ b/tools/cfgs/dataset_configs/custom_dataset.yaml @@ -0,0 +1,70 @@ +DATASET: 'CustomDataset' +DATA_PATH: '../data/custom' + +POINT_CLOUD_RANGE: [-75.2, -75.2, -2, 75.2, 75.2, 4] + +MAP_CLASS_TO_KITTI: { + 'Vehicle': 'Car', + 'Pedestrian': 'Pedestrian', + 'Cyclist': 'Cyclist', +} + +DATA_SPLIT: { + 'train': train, + 'test': val +} + +INFO_PATH: { + 'train': [custom_infos_train.pkl], + 'test': [custom_infos_val.pkl], +} + +POINT_FEATURE_ENCODING: { + encoding_type: absolute_coordinates_encoding, + used_feature_list: ['x', 'y', 'z', 'intensity'], + src_feature_list: ['x', 'y', 'z', 'intensity'], +} + +DATA_AUGMENTOR: + DISABLE_AUG_LIST: ['placeholder'] + AUG_CONFIG_LIST: + - NAME: gt_sampling + USE_ROAD_PLANE: False + DB_INFO_PATH: + - custom_dbinfos_train.pkl + PREPARE: { + filter_by_min_points: ['Vehicle:5', 'Pedestrian:5', 'Cyclist:5'], + } + + SAMPLE_GROUPS: ['Vehicle:20', 'Pedestrian:15', 'Cyclist:15'] + NUM_POINT_FEATURES: 4 + DATABASE_WITH_FAKELIDAR: False + REMOVE_EXTRA_WIDTH: [0.0, 0.0, 0.0] + LIMIT_WHOLE_SCENE: True + + - NAME: random_world_flip + ALONG_AXIS_LIST: ['x', 'y'] + + - NAME: random_world_rotation + WORLD_ROT_ANGLE: [-0.78539816, 0.78539816] + + - NAME: random_world_scaling + WORLD_SCALE_RANGE: [0.95, 1.05] + +DATA_PROCESSOR: + - NAME: mask_points_and_boxes_outside_range + REMOVE_OUTSIDE_BOXES: True + + - NAME: shuffle_points + SHUFFLE_ENABLED: { + 'train': True, + 'test': False + } + + - NAME: transform_points_to_voxels + VOXEL_SIZE: [0.1, 0.1, 0.15] + MAX_POINTS_PER_VOXEL: 5 + MAX_NUMBER_OF_VOXELS: { + 'train': 150000, + 'test': 150000 + } \ No newline at end of file