Add/update pointpainting
Browse files- pointpainting/deploy_metadata.yaml +1 -0
- pointpainting/detection_class_remapper.param.yaml +38 -0
- pointpainting/meta.json +82 -0
- pointpainting/pointpainting_ml_package.param.yaml +24 -0
- pointpainting/pts_backbone_neck_head_pointpainting.onnx +3 -0
- pointpainting/pts_voxel_encoder_pointpainting.onnx +3 -0
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
|