Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenPCDet
Commits
55848d9f
Unverified
Commit
55848d9f
authored
Nov 18, 2022
by
CSautier
Committed by
GitHub
Nov 18, 2022
Browse files
ONCE CenterPoint merge (#1189)
* New CenterPoint parameters and scores
parent
707a861a
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
38 deletions
+2
-38
README.md
README.md
+2
-1
tools/cfgs/once_models/centerpoint.yaml
tools/cfgs/once_models/centerpoint.yaml
+0
-37
No files found.
README.md
View file @
55848d9f
...
...
@@ -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.
...
...
tools/cfgs/once_models/centerpoint.yaml
View file @
55848d9f
...
...
@@ -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
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment