|
@@ -1,9 +1,11 @@
|
|
|
-CLASS_NAMES: ['car','truck', 'construction_vehicle', 'bus', 'trailer',
|
|
|
- 'barrier', 'motorcycle', 'bicycle', 'pedestrian', 'traffic_cone']
|
|
|
+#CLASS_NAMES: ['轿车', '卡车', '施工车辆', '公交车', '拖车/挂车',
|
|
|
+# '路碍', '摩托车', '自行车', '行人', '锥形筒']
|
|
|
|
|
|
+CLASS_NAMES: ['car', 'truck', 'bus', 'barrier', 'bicycle', 'pedestrian', 'traffic_cone']
|
|
|
DATA_CONFIG:
|
|
|
- _BASE_CONFIG_: cfgs/dataset_configs/nuscenes_dataset.yaml
|
|
|
+ _BASE_CONFIG_: /home/a/tian/OpenPCDet/tools/cfgs/dataset_configs/nuscenes_dataset_new.yaml
|
|
|
|
|
|
+ #POINT_CLOUD_RANGE: [-38.4, -38.4, -5.0, 38.4, 38.4, 3.0]
|
|
|
POINT_CLOUD_RANGE: [-51.2, -51.2, -5.0, 51.2, 51.2, 3.0]
|
|
|
DATA_PROCESSOR:
|
|
|
- NAME: mask_points_and_boxes_outside_range
|
|
@@ -58,9 +60,9 @@ MODEL:
|
|
|
ANCHOR_GENERATOR_CONFIG: [
|
|
|
{
|
|
|
'class_name': car,
|
|
|
- 'anchor_sizes': [[4.63, 1.97, 1.74]],
|
|
|
+ 'anchor_sizes': [[4.62, 1.95, 1.73]],
|
|
|
'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-0.95],
|
|
|
+ 'anchor_bottom_heights': [-0.85],
|
|
|
'align_center': False,
|
|
|
'feature_map_stride': 4,
|
|
|
'matched_threshold': 0.6,
|
|
@@ -68,79 +70,49 @@ MODEL:
|
|
|
},
|
|
|
{
|
|
|
'class_name': truck,
|
|
|
- 'anchor_sizes': [[6.93, 2.51, 2.84]],
|
|
|
+ 'anchor_sizes': [[6.85, 2.56, 2.89]],
|
|
|
'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-0.6],
|
|
|
+ 'anchor_bottom_heights': [-0.23],
|
|
|
'align_center': False,
|
|
|
'feature_map_stride': 4,
|
|
|
- 'matched_threshold': 0.55,
|
|
|
- 'unmatched_threshold': 0.4
|
|
|
- },
|
|
|
- {
|
|
|
- 'class_name': construction_vehicle,
|
|
|
- 'anchor_sizes': [[6.37, 2.85, 3.19]],
|
|
|
- 'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-0.225],
|
|
|
- 'align_center': False,
|
|
|
- 'feature_map_stride': 4,
|
|
|
- 'matched_threshold': 0.5,
|
|
|
- 'unmatched_threshold': 0.35
|
|
|
+ 'matched_threshold': 0.525,
|
|
|
+ 'unmatched_threshold': 0.375
|
|
|
},
|
|
|
{
|
|
|
'class_name': bus,
|
|
|
- 'anchor_sizes': [[10.5, 2.94, 3.47]],
|
|
|
- 'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-0.085],
|
|
|
- 'align_center': False,
|
|
|
- 'feature_map_stride': 4,
|
|
|
- 'matched_threshold': 0.55,
|
|
|
- 'unmatched_threshold': 0.4
|
|
|
- },
|
|
|
- {
|
|
|
- 'class_name': trailer,
|
|
|
- 'anchor_sizes': [[12.29, 2.90, 3.87]],
|
|
|
+ 'anchor_sizes': [[11.81, 2.91, 3.71]],
|
|
|
'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [0.115],
|
|
|
+ 'anchor_bottom_heights': [0.24],
|
|
|
'align_center': False,
|
|
|
'feature_map_stride': 4,
|
|
|
- 'matched_threshold': 0.5,
|
|
|
- 'unmatched_threshold': 0.35
|
|
|
+ 'matched_threshold': 0.525,
|
|
|
+ 'unmatched_threshold': 0.375
|
|
|
},
|
|
|
{
|
|
|
'class_name': barrier,
|
|
|
'anchor_sizes': [[0.50, 2.53, 0.98]],
|
|
|
'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-1.33],
|
|
|
+ 'anchor_bottom_heights': [-1.18],
|
|
|
'align_center': False,
|
|
|
'feature_map_stride': 4,
|
|
|
'matched_threshold': 0.55,
|
|
|
'unmatched_threshold': 0.4
|
|
|
},
|
|
|
- {
|
|
|
- 'class_name': motorcycle,
|
|
|
- 'anchor_sizes': [[2.11, 0.77, 1.47]],
|
|
|
- 'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-1.085],
|
|
|
- 'align_center': False,
|
|
|
- 'feature_map_stride': 4,
|
|
|
- 'matched_threshold': 0.5,
|
|
|
- 'unmatched_threshold': 0.3
|
|
|
- },
|
|
|
{
|
|
|
'class_name': bicycle,
|
|
|
- 'anchor_sizes': [[1.70, 0.60, 1.28]],
|
|
|
+ 'anchor_sizes': [[1.91, 0.69, 1.38]],
|
|
|
'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-1.18],
|
|
|
+ 'anchor_bottom_heights': [-0.96],
|
|
|
'align_center': False,
|
|
|
'feature_map_stride': 4,
|
|
|
'matched_threshold': 0.5,
|
|
|
- 'unmatched_threshold': 0.35
|
|
|
+ 'unmatched_threshold': 0.325
|
|
|
},
|
|
|
{
|
|
|
'class_name': pedestrian,
|
|
|
- 'anchor_sizes': [[0.73, 0.67, 1.77]],
|
|
|
+ 'anchor_sizes': [[0.72, 0.67, 1.77]],
|
|
|
'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-0.935],
|
|
|
+ 'anchor_bottom_heights': [-0.67],
|
|
|
'align_center': False,
|
|
|
'feature_map_stride': 4,
|
|
|
'matched_threshold': 0.6,
|
|
@@ -150,36 +122,13 @@ MODEL:
|
|
|
'class_name': traffic_cone,
|
|
|
'anchor_sizes': [[0.41, 0.41, 1.07]],
|
|
|
'anchor_rotations': [0, 1.57],
|
|
|
- 'anchor_bottom_heights': [-1.285],
|
|
|
+ 'anchor_bottom_heights': [-1.23],
|
|
|
'align_center': False,
|
|
|
'feature_map_stride': 4,
|
|
|
'matched_threshold': 0.6,
|
|
|
'unmatched_threshold': 0.4
|
|
|
},
|
|
|
]
|
|
|
-
|
|
|
- SHARED_CONV_NUM_FILTER: 64
|
|
|
-
|
|
|
- RPN_HEAD_CFGS: [
|
|
|
- {
|
|
|
- 'HEAD_CLS_NAME': ['car'],
|
|
|
- },
|
|
|
- {
|
|
|
- 'HEAD_CLS_NAME': ['truck', 'construction_vehicle'],
|
|
|
- },
|
|
|
- {
|
|
|
- 'HEAD_CLS_NAME': ['bus', 'trailer'],
|
|
|
- },
|
|
|
- {
|
|
|
- 'HEAD_CLS_NAME': ['barrier'],
|
|
|
- },
|
|
|
- {
|
|
|
- 'HEAD_CLS_NAME': ['motorcycle', 'bicycle'],
|
|
|
- },
|
|
|
- {
|
|
|
- 'HEAD_CLS_NAME': ['pedestrian', 'traffic_cone'],
|
|
|
- },
|
|
|
- ]
|
|
|
SEPARATE_REG_CONFIG:
|
|
|
NUM_MIDDLE_CONV: 1
|
|
|
NUM_MIDDLE_FILTER: 64
|
|
@@ -226,7 +175,7 @@ MODEL:
|
|
|
|
|
|
OPTIMIZATION:
|
|
|
BATCH_SIZE_PER_GPU: 4
|
|
|
- NUM_EPOCHS: 20
|
|
|
+ NUM_EPOCHS: 30
|
|
|
|
|
|
OPTIMIZER: adam_onecycle
|
|
|
LR: 0.001
|
|
@@ -235,7 +184,7 @@ OPTIMIZATION:
|
|
|
|
|
|
MOMS: [0.95, 0.85]
|
|
|
PCT_START: 0.4
|
|
|
- DIV_FACTOR: 10
|
|
|
+ DIV_FACTOR: 20
|
|
|
DECAY_STEP_LIST: [35, 45]
|
|
|
LR_DECAY: 0.1
|
|
|
LR_CLIP: 0.0000001
|
|
@@ -244,3 +193,4 @@ OPTIMIZATION:
|
|
|
WARMUP_EPOCH: 1
|
|
|
|
|
|
GRAD_NORM_CLIP: 10
|
|
|
+
|