Upload 13 files
Browse files- .gitattributes +2 -0
- centerpoint_ml_package.param.yaml +23 -0
- centerpoint_sigma_ml_package.param.yaml +23 -0
- centerpoint_tiny_ml_package.param.yaml +23 -0
- deploy_metadata.yaml +1 -0
- detection_class_remapper.param.yaml +38 -0
- pts_backbone_neck_head_centerpoint.onnx +3 -0
- pts_backbone_neck_head_centerpoint_tiny.engine +3 -0
- pts_backbone_neck_head_centerpoint_tiny.json +43 -0
- pts_backbone_neck_head_centerpoint_tiny.onnx +3 -0
- pts_voxel_encoder_centerpoint.onnx +3 -0
- pts_voxel_encoder_centerpoint_tiny.engine +3 -0
- pts_voxel_encoder_centerpoint_tiny.json +13 -0
- pts_voxel_encoder_centerpoint_tiny.onnx +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
pts_backbone_neck_head_centerpoint_tiny.engine filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
pts_voxel_encoder_centerpoint_tiny.engine filter=lfs diff=lfs merge=lfs -text
|
centerpoint_ml_package.param.yaml
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
/**:
|
| 2 |
+
ros__parameters:
|
| 3 |
+
model_params:
|
| 4 |
+
class_names: ["CAR", "TRUCK", "BUS", "BICYCLE", "PEDESTRIAN"]
|
| 5 |
+
point_feature_size: 4
|
| 6 |
+
max_voxel_size: 40000
|
| 7 |
+
point_cloud_range: [-76.8, -76.8, -4.0, 76.8, 76.8, 6.0]
|
| 8 |
+
voxel_size: [0.32, 0.32, 10.0]
|
| 9 |
+
downsample_factor: 1
|
| 10 |
+
encoder_in_feature_size: 9
|
| 11 |
+
has_variance: false
|
| 12 |
+
has_twist: false
|
| 13 |
+
detection_score_thresholds:
|
| 14 |
+
# Upper bound of radial distance of 3d center_x and center_y in bounding boxes, and starting from 0
|
| 15 |
+
# [0-50m), [50.0-90m), [90.0-121.0m), [121.0-200.0m), set a very high number for the last one to be safe
|
| 16 |
+
distance_bin_upper_limits: [50.0, 90.0, 121.0, 200.0]
|
| 17 |
+
# Each list of thresholds must match with the number of upper_bounds, and each threshold corresponds to a distance bucket in the order of the upper bounds
|
| 18 |
+
min_confidence_scores:
|
| 19 |
+
CAR: [0.35, 0.35, 0.35, 0.35]
|
| 20 |
+
TRUCK: [0.35, 0.35, 0.35, 0.35]
|
| 21 |
+
BUS: [0.35, 0.35, 0.35, 0.35]
|
| 22 |
+
BICYCLE: [0.35, 0.35, 0.35, 0.35]
|
| 23 |
+
PEDESTRIAN: [0.35, 0.35, 0.35, 0.35]
|
centerpoint_sigma_ml_package.param.yaml
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
/**:
|
| 2 |
+
ros__parameters:
|
| 3 |
+
model_params:
|
| 4 |
+
class_names: ["CAR", "TRUCK", "BUS", "BICYCLE", "PEDESTRIAN"]
|
| 5 |
+
point_feature_size: 4
|
| 6 |
+
max_voxel_size: 40000
|
| 7 |
+
point_cloud_range: [-76.8, -76.8, -4.0, 76.8, 76.8, 6.0]
|
| 8 |
+
voxel_size: [0.32, 0.32, 10.0]
|
| 9 |
+
downsample_factor: 1
|
| 10 |
+
encoder_in_feature_size: 9
|
| 11 |
+
has_variance: false
|
| 12 |
+
has_twist: false
|
| 13 |
+
detection_score_thresholds:
|
| 14 |
+
# Upper bound of radial distance of 3d center_x and center_y in bounding boxes, and starting from 0
|
| 15 |
+
# [0-50m), [50.0-90m), [90.0-121.0m), [121.0-200.0m), set a very high number for the last one to be safe
|
| 16 |
+
distance_bin_upper_limits: [50.0, 90.0, 121.0, 200.0]
|
| 17 |
+
# Each list of thresholds must match with the number of upper_bounds, and each threshold corresponds to a distance bucket in the order of the upper bounds
|
| 18 |
+
min_confidence_scores:
|
| 19 |
+
CAR: [0.35, 0.35, 0.35, 0.35]
|
| 20 |
+
TRUCK: [0.35, 0.35, 0.35, 0.35]
|
| 21 |
+
BUS: [0.35, 0.35, 0.35, 0.35]
|
| 22 |
+
BICYCLE: [0.35, 0.35, 0.35, 0.35]
|
| 23 |
+
PEDESTRIAN: [0.35, 0.35, 0.35, 0.35]
|
centerpoint_tiny_ml_package.param.yaml
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
/**:
|
| 2 |
+
ros__parameters:
|
| 3 |
+
model_params:
|
| 4 |
+
class_names: ["CAR", "TRUCK", "BUS", "BICYCLE", "PEDESTRIAN"]
|
| 5 |
+
point_feature_size: 4
|
| 6 |
+
max_voxel_size: 40000
|
| 7 |
+
point_cloud_range: [-76.8, -76.8, -4.0, 76.8, 76.8, 6.0]
|
| 8 |
+
voxel_size: [0.32, 0.32, 10.0]
|
| 9 |
+
downsample_factor: 2
|
| 10 |
+
encoder_in_feature_size: 9
|
| 11 |
+
has_variance: false
|
| 12 |
+
has_twist: false
|
| 13 |
+
detection_score_thresholds:
|
| 14 |
+
# Upper bound of radial distance of 3d center_x and center_y in bounding boxes, and starting from 0
|
| 15 |
+
# [0-50m), [50.0-90m), [90.0-121.0m), [121.0-200.0m), set a very high number for the last one to be safe
|
| 16 |
+
distance_bin_upper_limits: [50.0, 90.0, 121.0, 200.0]
|
| 17 |
+
# Each list of thresholds must match with the number of upper_bounds, and each threshold corresponds to a distance bucket in the order of the upper bounds
|
| 18 |
+
min_confidence_scores:
|
| 19 |
+
CAR: [0.35, 0.35, 0.35, 0.35]
|
| 20 |
+
TRUCK: [0.35, 0.35, 0.35, 0.35]
|
| 21 |
+
BUS: [0.35, 0.35, 0.35, 0.35]
|
| 22 |
+
BICYCLE: [0.35, 0.35, 0.35, 0.35]
|
| 23 |
+
PEDESTRIAN: [0.35, 0.35, 0.35, 0.35]
|
deploy_metadata.yaml
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
num_checkpoint: 9
|
detection_class_remapper.param.yaml
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
/**:
|
| 2 |
+
ros__parameters:
|
| 3 |
+
allow_remapping_by_area_matrix:
|
| 4 |
+
# NOTE(kl): We turn all vehicles into trailers if they go over 3x12 [m^2].
|
| 5 |
+
# NOTE(kl): We turn cars into trucks if they have an area between 2.2 x 5.5 and 3.0 * 12.0 [m^2]
|
| 6 |
+
# row: original class. column: class to remap to
|
| 7 |
+
#UNKNOWN, CAR, TRUCK, BUS, TRAILER, MOTORBIKE, BICYCLE,PEDESTRIAN
|
| 8 |
+
[0, 0, 0, 0, 0, 0, 0, 0, #UNKNOWN
|
| 9 |
+
0, 0, 1, 0, 1, 0, 0, 0, #CAR
|
| 10 |
+
0, 0, 0, 0, 1, 0, 0, 0, #TRUCK
|
| 11 |
+
0, 0, 0, 0, 1, 0, 0, 0, #BUS
|
| 12 |
+
0, 0, 0, 0, 0, 0, 0, 0, #TRAILER
|
| 13 |
+
0, 0, 0, 0, 0, 0, 0, 0, #MOTORBIKE
|
| 14 |
+
0, 0, 0, 0, 0, 0, 0, 0, #BICYCLE
|
| 15 |
+
0, 0, 0, 0, 0, 0, 0, 0] #PEDESTRIAN
|
| 16 |
+
|
| 17 |
+
min_area_matrix:
|
| 18 |
+
#UNKNOWN, CAR, TRUCK, BUS, TRAILER, MOTORBIKE, BICYCLE, PEDESTRIAN
|
| 19 |
+
[ 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #UNKNOWN
|
| 20 |
+
0.000, 0.000, 12.100, 0.000, 36.000, 0.000, 0.000, 0.000, #CAR
|
| 21 |
+
0.000, 0.000, 0.000, 0.000, 36.000, 0.000, 0.000, 0.000, #TRUCK
|
| 22 |
+
0.000, 0.000, 0.000, 0.000, 36.000, 0.000, 0.000, 0.000, #BUS
|
| 23 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #TRAILER
|
| 24 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #MOTORBIKE
|
| 25 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #BICYCLE
|
| 26 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000] #PEDESTRIAN
|
| 27 |
+
|
| 28 |
+
|
| 29 |
+
max_area_matrix:
|
| 30 |
+
#UNKNOWN, CAR, TRUCK, BUS, TRAILER, MOTORBIKE, BICYCLE, PEDESTRIAN
|
| 31 |
+
[ 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #UNKNOWN
|
| 32 |
+
0.000, 0.000, 36.000, 0.000, 999.999, 0.000, 0.000, 0.000, #CAR
|
| 33 |
+
0.000, 0.000, 0.000, 0.000, 999.999, 0.000, 0.000, 0.000, #TRUCK
|
| 34 |
+
0.000, 0.000, 0.000, 0.000, 999.999, 0.000, 0.000, 0.000, #BUS
|
| 35 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #TRAILER
|
| 36 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #MOTORBIKE
|
| 37 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, #BICYCLE
|
| 38 |
+
0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000] #PEDESTRIAN
|
pts_backbone_neck_head_centerpoint.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3fe7e128955646740c41a25be0c8f141d5a94594fe79d7405fe2a859e391542e
|
| 3 |
+
size 20140082
|
pts_backbone_neck_head_centerpoint_tiny.engine
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4b92ee03f59e0f79f23962e4f837cdfe0a939ba6e475cb13589ab919a71faedf
|
| 3 |
+
size 11959444
|
pts_backbone_neck_head_centerpoint_tiny.json
ADDED
|
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"Layers": ["Reformatting CopyNode for Input Tensor 0 to Conv_0 + Relu_1"
|
| 2 |
+
,"Conv_0 + Relu_1"
|
| 3 |
+
,"Conv_2 + Relu_3"
|
| 4 |
+
,"Conv_4 + Relu_5"
|
| 5 |
+
,"Conv_6 + Relu_7"
|
| 6 |
+
,"Conv_8 + Relu_9"
|
| 7 |
+
,"Conv_10 + Relu_11"
|
| 8 |
+
,"Conv_12 + Relu_13"
|
| 9 |
+
,"Conv_14 + Relu_15"
|
| 10 |
+
,"Conv_16 + Relu_17"
|
| 11 |
+
,"Conv_18 + Relu_19"
|
| 12 |
+
,"Conv_20 + Relu_21"
|
| 13 |
+
,"Conv_22 + Relu_23"
|
| 14 |
+
,"Conv_24 + Relu_25"
|
| 15 |
+
,"Conv_26 + Relu_27"
|
| 16 |
+
,"Conv_28 + Relu_29"
|
| 17 |
+
,"Conv_30 + Relu_31"
|
| 18 |
+
,"Conv_32 + Relu_33"
|
| 19 |
+
,"ConvTranspose_34 + BatchNormalization_35 + Relu_36"
|
| 20 |
+
,"ConvTranspose_37 + BatchNormalization_38 + Relu_39"
|
| 21 |
+
,"Conv_41 + Relu_42"
|
| 22 |
+
,"Conv_43 + Relu_44 || Conv_46 + Relu_47 || Conv_49 + Relu_50 || Conv_52 + Relu_53 || Conv_55 + Relu_56 || Conv_58 + Relu_59"
|
| 23 |
+
,"Conv_45"
|
| 24 |
+
,"Reformatting CopyNode for Output Tensor 0 to Conv_45"
|
| 25 |
+
,"Conv_48"
|
| 26 |
+
,"Reformatting CopyNode for Output Tensor 0 to Conv_48"
|
| 27 |
+
,"Conv_51"
|
| 28 |
+
,"Reformatting CopyNode for Output Tensor 0 to Conv_51"
|
| 29 |
+
,"Conv_54"
|
| 30 |
+
,"Reformatting CopyNode for Output Tensor 0 to Conv_54"
|
| 31 |
+
,"Conv_57"
|
| 32 |
+
,"Reformatting CopyNode for Output Tensor 0 to Conv_57"
|
| 33 |
+
,"Conv_60"
|
| 34 |
+
,"Reformatting CopyNode for Output Tensor 0 to Conv_60"
|
| 35 |
+
],
|
| 36 |
+
"Bindings": ["spatial_features"
|
| 37 |
+
,"heatmap"
|
| 38 |
+
,"reg"
|
| 39 |
+
,"height"
|
| 40 |
+
,"dim"
|
| 41 |
+
,"rot"
|
| 42 |
+
,"vel"
|
| 43 |
+
]}
|
pts_backbone_neck_head_centerpoint_tiny.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9bb0b634f3664bd098ce7d6a3d8a9fb7cc8d9b8252b27f302c71e43316bab551
|
| 3 |
+
size 20140082
|
pts_voxel_encoder_centerpoint.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dc1a876580d86ee7a341d543f8ade2ede7f43bd032dc5b44155b1f0175405764
|
| 3 |
+
size 7429
|
pts_voxel_encoder_centerpoint_tiny.engine
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7ad4303b2a180ab79c1e86653f70487d2c1cff1b2a71c08c8648c907d6810318
|
| 3 |
+
size 18127764
|
pts_voxel_encoder_centerpoint_tiny.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"Layers": ["Reformatting CopyNode for Input Tensor 0 to {ForeignNode[45 + ONNXTRT_Broadcast...ReduceMax_23]}"
|
| 2 |
+
,"dummy_shape_call__mye9612_0_myl1_0"
|
| 3 |
+
,"__myl_MovCon_myl1_1"
|
| 4 |
+
,"MatMul_0_myl1_2"
|
| 5 |
+
,"__myl_TraResMulAddResMovTraRelMax_myl1_3"
|
| 6 |
+
,"__myl_RepCon_myl1_4"
|
| 7 |
+
,"MatMul_18_myl1_5"
|
| 8 |
+
,"__myl_TraResMulAddResMovTraRelMax_myl1_6"
|
| 9 |
+
,"Reformatting CopyNode for Output Tensor 0 to {ForeignNode[45 + ONNXTRT_Broadcast...ReduceMax_23]}"
|
| 10 |
+
],
|
| 11 |
+
"Bindings": ["input_features"
|
| 12 |
+
,"pillar_features"
|
| 13 |
+
]}
|
pts_voxel_encoder_centerpoint_tiny.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2c53465715c1fd2e9dc5727ef3fca74f4cdf0538f74286b0946e219d0ca5693b
|
| 3 |
+
size 7429
|