pshashid commited on
Commit
422e685
·
verified ·
1 Parent(s): f5f9f17

Add/update pointpainting

Browse files
pointpainting/deploy_metadata.yaml ADDED
@@ -0,0 +1 @@
 
 
1
+ version: v5.0
pointpainting/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
pointpainting/meta.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "pointpainting",
3
+ "source": "AutowareFoundation/image_projection_based_fusion",
4
+ "method": "prebuilt",
5
+ "note": null,
6
+ "revision": "33119d71afd53fbc3994dcffc71e6247b76cdd5a",
7
+ "validation": "prebuilt, not re-validated",
8
+ "opset_requested": 17,
9
+ "files": {
10
+ "pts_backbone_neck_head_pointpainting.onnx": {
11
+ "sha256": "7fe62fcebe0e0f62a000d06aa94d779feb444d933671a4a3189fe01be8c19a00",
12
+ "bytes": 20140082,
13
+ "inputs": {
14
+ "spatial_features": [
15
+ "batch_size",
16
+ 32,
17
+ "H",
18
+ "W"
19
+ ]
20
+ },
21
+ "outputs": {
22
+ "heatmap": [
23
+ "batch_size",
24
+ 5,
25
+ "H",
26
+ "W"
27
+ ],
28
+ "reg": [
29
+ "batch_size",
30
+ 2,
31
+ "H",
32
+ "W"
33
+ ],
34
+ "height": [
35
+ "batch_size",
36
+ 1,
37
+ "H",
38
+ "W"
39
+ ],
40
+ "dim": [
41
+ "batch_size",
42
+ 3,
43
+ "H",
44
+ "W"
45
+ ],
46
+ "rot": [
47
+ "batch_size",
48
+ 2,
49
+ "H",
50
+ "W"
51
+ ],
52
+ "vel": [
53
+ "batch_size",
54
+ 2,
55
+ "H",
56
+ "W"
57
+ ]
58
+ },
59
+ "opset": 11
60
+ },
61
+ "pts_voxel_encoder_pointpainting.onnx": {
62
+ "sha256": "3ca452ea5ca9467bf782955f75704ba8466841e275e8b8acd991b9911d53249e",
63
+ "bytes": 7621,
64
+ "inputs": {
65
+ "input_features": [
66
+ "num_voxels",
67
+ "num_max_points",
68
+ 12
69
+ ]
70
+ },
71
+ "outputs": {
72
+ "pillar_features": [
73
+ "num_voxels",
74
+ 1,
75
+ 32
76
+ ]
77
+ },
78
+ "opset": 11
79
+ }
80
+ },
81
+ "exported_at": "2026-10-19T18:45:37.492531+00:00"
82
+ }
pointpainting/pointpainting_ml_package.param.yaml ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ /**:
2
+ ros__parameters:
3
+ model_params:
4
+ class_names: ["CAR", "TRUCK", "BUS", "BICYCLE", "PEDESTRIAN"]
5
+ paint_class_names: ["CAR", "BICYCLE", "PEDESTRIAN"]
6
+ point_feature_size: 7 # x, y, z, time-lag and car, pedestrian, bicycle
7
+ max_voxel_size: 40000
8
+ point_cloud_range: [-121.6, -76.8, -3.0, 121.6, 76.8, 5.0]
9
+ voxel_size: [0.32, 0.32, 8.0]
10
+ downsample_factor: 1
11
+ encoder_in_feature_size: 12
12
+ has_twist: false
13
+ has_variance: false
14
+ detection_score_thresholds:
15
+ # Upper bound of radial distance of 3d center_x and center_y in bounding boxes, and starting from 0
16
+ # [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
17
+ distance_bin_upper_limits: [50.0, 90.0, 121.0, 200.0]
18
+ # 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
19
+ min_confidence_scores:
20
+ CAR: [0.35, 0.35, 0.35, 0.35]
21
+ TRUCK: [0.35, 0.35, 0.35, 0.35]
22
+ BUS: [0.35, 0.35, 0.35, 0.35]
23
+ BICYCLE: [0.35, 0.35, 0.35, 0.35]
24
+ PEDESTRIAN: [0.35, 0.35, 0.35, 0.35]
pointpainting/pts_backbone_neck_head_pointpainting.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fe62fcebe0e0f62a000d06aa94d779feb444d933671a4a3189fe01be8c19a00
3
+ size 20140082
pointpainting/pts_voxel_encoder_pointpainting.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ca452ea5ca9467bf782955f75704ba8466841e275e8b8acd991b9911d53249e
3
+ size 7621