Unverified Commit 55848d9f authored by CSautier's avatar CSautier Committed by GitHub
Browse files

ONCE CenterPoint merge (#1189)

* New CenterPoint parameters and scores
parent 707a861a
......@@ -216,7 +216,8 @@ All models are trained with 8 GPUs.
| [PointPillar](tools/cfgs/once_models/pointpillar.yaml) | 68.57 | 17.63 | 46.81 | 44.34 |
| [SECOND](tools/cfgs/once_models/second.yaml) | 71.19 | 26.44 | 58.04 | 51.89 |
| [PV-RCNN](tools/cfgs/once_models/pv_rcnn.yaml) | 77.77 | 23.50 | 59.37 | 53.55 |
| [CenterPoint](tools/cfgs/once_models/centerpoint.yaml) | 66.79 | 49.90 | 63.45 | 60.05 |
| [CenterPoint](tools/cfgs/once_models/centerpoint.yaml) | 78.02 | 49.74 | 67.22 | 64.99 |
*Results are not reproductible with this version of the code
### Other datasets
Welcome to support other datasets by submitting pull request.
......
......@@ -45,57 +45,20 @@ MODEL:
'rot': {'out_channels': 2, 'num_conv': 2},
}
# DATASET: once
# MODE: 3d
TARGET_ASSIGNER_CONFIG:
FEATURE_MAP_STRIDE: 8
NUM_MAX_OBJS: 500
GAUSSIAN_OVERLAP: 0.1
MIN_RADIUS: 2
# tasks: *tasks_head
DENSE_REG: 1
# mapping: {
# "Car": 1,
# "Bus": 2,
# "Truck": 3,
# "Pedestrian": 4,
# "Cyclist": 5
# }
LOSS_CONFIG:
LOSS_WEIGHTS: {
'cls_weight': 1.0,
'loc_weight': 1.0,
# weight: 0.25
'code_weights': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
}
# TEST_CONFIG:
# pc_range: [-75.2, -75.2]
# out_size_factor: 8
# voxel_size: [0.1, 0.1]
# nms:
# train:
# use_iou_3d_nms: True
# use_rotate_nms: False
# use_maxpool_nms: False
# use_circle_nms: False
# min_radius: [4, 10, 12, 0.175, 0.85]
# nms_iou_threshold: 0.8
# nms_pre_max_size: 1500
# nms_post_max_size: 80
# test:
# use_iou_3d_nms: True
# use_rotate_nms: False
# use_maxpool_nms: False
# use_circle_nms: False
# min_radius: [4, 10, 12, 0.175, 0.85]
# nms_iou_threshold: 0.01
# nms_pre_max_size: 500
# nms_post_max_size: 83
POST_PROCESSING:
SCORE_THRESH: 0.1
POST_CENTER_LIMIT_RANGE: [-75.2, -75.2, -5.0, 75.2, 75.2, 3.0]
......
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