Object Detection
ONNX
xmfcx commited on
Commit
c7de0b1
·
verified ·
1 Parent(s): d0a58e5

Upload 13 files

Browse files
.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