Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
91 commits
Select commit Hold shift + click to select a range
9bb04ce
Update docker installation commit for perception_evaluation
KSeangTan Jan 14, 2026
ebee4a0
Merge branch 'tier4:main' into main
KSeangTan Jan 14, 2026
775f1d0
Resolve merge conflicts
KSeangTan Jan 20, 2026
edb373a
Merge branch 'tier4:main' into main
KSeangTan Jan 20, 2026
a358886
Merge branch 'tier4:main' into main
KSeangTan Jan 20, 2026
be329c4
Merge branch 'tier4:main' into main
KSeangTan Jan 21, 2026
6b81116
Merge branch 'tier4:main' into main
KSeangTan Jan 23, 2026
3e3c09d
Merge branch 'tier4:main' into main
KSeangTan Jan 29, 2026
fdeaa33
Merge branch 'tier4:main' into main
KSeangTan Feb 9, 2026
9b71536
Merge branch 'tier4:main' into main
KSeangTan Feb 9, 2026
3b3c734
Merge branch 'tier4:main' into main
KSeangTan Feb 17, 2026
6165a16
Merge branch 'tier4:main' into main
KSeangTan Feb 27, 2026
6d20865
Merge branch 'tier4:main' into main
KSeangTan Feb 27, 2026
0079f71
Merge branch 'tier4:main' into main
KSeangTan Mar 3, 2026
9a1130b
Merge branch 'tier4:main' into main
KSeangTan Mar 4, 2026
94ab3a3
Merge branch 'tier4:main' into main
KSeangTan Mar 6, 2026
cfd5fac
Merge branch 'tier4:main' into main
KSeangTan Mar 17, 2026
e57b46f
Merge branch 'tier4:main' into main
KSeangTan Mar 18, 2026
1a7ebee
Update config
KSeangTan Mar 19, 2026
8402451
Update config
KSeangTan Mar 24, 2026
cf6e13d
Update config
KSeangTan Mar 25, 2026
e1ebc99
Update config
KSeangTan Mar 26, 2026
2dac433
Update config
KSeangTan Mar 27, 2026
9046b7b
Update config
KSeangTan Mar 27, 2026
d4d93fa
Added
KSeangTan Mar 27, 2026
b2714a8
Added
KSeangTan Mar 31, 2026
43adb38
ci(pre-commit): autofix
pre-commit-ci[bot] Mar 31, 2026
73dd1c4
Added
KSeangTan Apr 8, 2026
89e2670
ci(pre-commit): autofix
pre-commit-ci[bot] Apr 8, 2026
dacc863
Merge branch 'tier4:main' into main
KSeangTan Apr 8, 2026
380d7aa
Added
KSeangTan Apr 10, 2026
b60e45a
Update configs
KSeangTan Apr 12, 2026
9b38a42
Added
KSeangTan Apr 13, 2026
f39b584
Updated
KSeangTan Apr 17, 2026
24a88ad
Added
KSeangTan Apr 17, 2026
c637420
Added
KSeangTan Apr 17, 2026
48879b8
ci(pre-commit): autofix
pre-commit-ci[bot] Apr 17, 2026
82457ab
Added
KSeangTan Apr 17, 2026
47d41e0
Added
KSeangTan Apr 17, 2026
827bbb2
Updated
KSeangTan Apr 17, 2026
5f9a4a5
Updated
KSeangTan Apr 17, 2026
a26782a
Updated
KSeangTan Apr 18, 2026
f4c01a5
Updated
KSeangTan Apr 18, 2026
15371af
Update dataset name
KSeangTan Apr 18, 2026
4977b33
Update dataset name
KSeangTan Apr 19, 2026
f85a8e9
Update dataset name
KSeangTan Apr 20, 2026
dd50b12
Merge branch 'tier4:main' into main
KSeangTan Apr 21, 2026
4222ab9
Update configs
KSeangTan Apr 21, 2026
0f5b588
Update configs
KSeangTan Apr 21, 2026
caecca6
Update base docstring
KSeangTan Apr 21, 2026
07c2e11
Update base docstring
KSeangTan Apr 21, 2026
2665b27
Update j6gen2_base and jpntaxi_base docstring
KSeangTan Apr 21, 2026
24d780b
Added
Apr 24, 2026
8175419
Added
Apr 24, 2026
33f11cd
Added
Apr 25, 2026
2237522
Added
Apr 25, 2026
60df4c0
Added
Apr 25, 2026
b5dabf2
Added
Apr 27, 2026
79024cf
Added
Apr 27, 2026
e771a69
Added
Apr 27, 2026
05703cb
Update configs
KSeangTan Apr 27, 2026
06ae9c2
Update configs
KSeangTan Apr 27, 2026
d8c1974
Update configs
KSeangTan Apr 28, 2026
25e581c
Update configs
KSeangTan Apr 28, 2026
9ba440a
Update configs
KSeangTan Apr 28, 2026
aac1e1d
Update configs
KSeangTan Apr 28, 2026
09c06d7
Update configs
KSeangTan Apr 28, 2026
5279b17
Update configs
KSeangTan Apr 28, 2026
9b4c2f2
Added
Apr 28, 2026
5c0ada8
Add the script
KSeangTan Apr 28, 2026
ad6b07a
Add the script
KSeangTan Apr 28, 2026
a592868
Add the script
KSeangTan Apr 28, 2026
2929ff6
Update configs
KSeangTan Apr 28, 2026
57d6ae6
Update configs
KSeangTan Apr 28, 2026
2e03655
Add the script
KSeangTan Apr 28, 2026
e2a69c1
Add the script
KSeangTan Apr 28, 2026
ebc8034
Update configs
KSeangTan Apr 28, 2026
bb35205
Add the script
KSeangTan Apr 29, 2026
f343fbe
Update configs
KSeangTan Apr 29, 2026
6f55027
Update configs
KSeangTan Apr 29, 2026
d99abd9
Add the script
KSeangTan Apr 30, 2026
b1b247a
Add the script
KSeangTan Apr 30, 2026
36e3811
Add the script
KSeangTan Apr 30, 2026
df3e50c
Merge branch 'tier4:main' into main
KSeangTan May 2, 2026
e905263
Add traffic cone and barrier
KSeangTan May 7, 2026
c429514
Merge branch 'main' into feat/add_traffic_barrier_to_dataset
KSeangTan May 7, 2026
dc2265e
ci(pre-commit): autofix
pre-commit-ci[bot] May 7, 2026
15bbf0e
remove unecessary changes
KSeangTan May 7, 2026
1323d4e
ci(pre-commit): autofix
pre-commit-ci[bot] May 7, 2026
585a0b2
remove unecessary changes
KSeangTan May 7, 2026
a47646e
ci(pre-commit): autofix
pre-commit-ci[bot] May 7, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -61,13 +61,15 @@ RUN python3 -m pip --no-cache-dir install \
RUN python3 -m pip install git+https://github.com/tier4/t4-devkit@v0.5.1

# Install autoware-perception-evaluation
RUN python3 -m pip install git+https://github.com/tier4/autoware_perception_evaluation@9d8c9773d35177bb0b7f2606f429f58a5fb708ca
RUN python3 -m pip install git+https://github.com/tier4/autoware_perception_evaluation@3c9577dc23fd76a049559b42656ca46c1c32fa66

# Need to dowgrade setuptools to 60.2.0 to fix setup
RUN python3 -m pip --no-cache-dir install \
setuptools==60.2.0 \
transformers==4.51.3 \
polars==1.37.1
polars==1.37.1 \
onnx_graphsurgeon==0.5.8 \
spconv-cu126==2.3.8

# NOTE(knzo25): this patch is needed to use numpy versions over 1.23.5 (version used in mmdet3d 1.4.0)
# It can be safely deleted when mmdet3d updates the numpy version
Expand Down
17 changes: 7 additions & 10 deletions autoware_ml/configs/detection3d/dataset/t4dataset/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,8 +91,8 @@
"pedestrian.stroller": "pedestrian",
"pedestrian.wheelchair": "pedestrian",
"movable_object.barrier": "barrier",
"movable_object.debris": "debris",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.debris": "barrier",
"movable_object.pushable_pullable": "barrier",
"movable_object.trafficcone": "traffic_cone",
"movable_object.traffic_cone": "traffic_cone",
"animal": "animal",
Expand All @@ -113,7 +113,7 @@
# DBv2.0 and DBv3.0
"animal": "animal",
"movable_object.barrier": "barrier",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.pushable_pullable": "barrier",
"movable_object.traffic_cone": "traffic_cone",
"pedestrian.adult": "pedestrian",
"pedestrian.child": "pedestrian",
Expand Down Expand Up @@ -143,15 +143,12 @@
"semi_trailer": "trailer",
"tractor_unit": "truck",
"construction_vehicle": "truck",
"traffic_cone": "traffic_cone",
"trafficcone": "traffic_cone",
"barrier": "barrier",
}

class_names = [
"car",
"truck",
"bus",
"bicycle",
"pedestrian",
]
class_names = ["car", "truck", "bus", "bicycle", "pedestrian", "traffic_cone", "barrier"]
num_class = len(class_names)
metainfo = dict(classes=class_names)

Expand Down
11 changes: 8 additions & 3 deletions autoware_ml/configs/detection3d/dataset/t4dataset/j6gen2.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,8 +72,8 @@
"pedestrian.stroller": "pedestrian",
"pedestrian.wheelchair": "pedestrian",
"movable_object.barrier": "barrier",
"movable_object.debris": "debris",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.debris": "barrier",
"movable_object.pushable_pullable": "barrier",
"movable_object.trafficcone": "traffic_cone",
"movable_object.traffic_cone": "traffic_cone",
"animal": "animal",
Expand All @@ -94,7 +94,7 @@
# DBv2.0 and DBv3.0
"animal": "animal",
"movable_object.barrier": "barrier",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.pushable_pullable": "barrier",
"movable_object.traffic_cone": "traffic_cone",
"pedestrian.adult": "pedestrian",
"pedestrian.child": "pedestrian",
Expand Down Expand Up @@ -124,6 +124,9 @@
"semi_trailer": "trailer",
"tractor_unit": "truck",
"construction_vehicle": "truck",
"traffic_cone": "traffic_cone",
"trafficcone": "traffic_cone",
"barrier": "barrier",
}

class_names = [
Expand All @@ -132,6 +135,8 @@
"bus",
"bicycle",
"pedestrian",
"traffic_cone",
"barrier",
]
num_class = len(class_names)
metainfo = dict(classes=class_names)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -78,8 +78,8 @@
"pedestrian.stroller": "pedestrian",
"pedestrian.wheelchair": "pedestrian",
"movable_object.barrier": "barrier",
"movable_object.debris": "debris",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.debris": "barrier",
"movable_object.pushable_pullable": "barrier",
"movable_object.trafficcone": "traffic_cone",
"movable_object.traffic_cone": "traffic_cone",
"animal": "animal",
Expand All @@ -100,7 +100,7 @@
# DBv2.0 and DBv3.0
"animal": "animal",
"movable_object.barrier": "barrier",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.pushable_pullable": "barrier",
"movable_object.traffic_cone": "traffic_cone",
"pedestrian.adult": "pedestrian",
"pedestrian.child": "pedestrian",
Expand Down Expand Up @@ -130,14 +130,20 @@
"semi_trailer": "trailer",
"tractor_unit": "truck",
"construction_vehicle": "truck",
"traffic_cone": "traffic_cone",
"trafficcone": "traffic_cone",
"barrier": "barrier",
}


class_names = [
"car",
"truck",
"bus",
"bicycle",
"pedestrian",
"traffic_cone",
"barrier",
]
num_class = len(class_names)
metainfo = dict(classes=class_names)
Expand Down
194 changes: 194 additions & 0 deletions autoware_ml/configs/detection3d/dataset/t4dataset/j6gen2_v2.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,194 @@
custom_imports = dict(
imports=[
"autoware_ml.detection3d.datasets.t4dataset",
"autoware_ml.detection3d.evaluation.t4metric.t4metric",
"autoware_ml.detection3d.evaluation.t4metric.t4metric_v2",
]
)

# dataset type setting
dataset_type = "T4Dataset"
info_train_file_name = "t4dataset_j6gen2_v2_infos_train.pkl"
info_val_file_name = "t4dataset_j6gen2_v2_infos_val.pkl"
info_test_file_name = "t4dataset_j6gen2_v2_infos_test.pkl"

info_train_statistics_file_name = "t4dataset_j6gen2_v2_statistics_train.parquet"
info_val_statistics_file_name = "t4dataset_j6gen2_v2_statistics_val.parquet"
info_test_statistics_file_name = "t4dataset_j6gen2_v2_statistics_test.parquet"

# dataset scene setting
dataset_version_list = [
"db_j6gen2_v2",
]

dataset_test_groups = {
"j6gen2_v2": ("t4dataset_j6gen2_v2_infos_test.pkl", True),
}

# dataset format setting
data_prefix = dict(
pts="",
CAM_FRONT="",
CAM_FRONT_LEFT="",
CAM_FRONT_RIGHT="",
CAM_BACK="",
CAM_BACK_RIGHT="",
CAM_BACK_LEFT="",
sweeps="",
)
camera_types = {
"CAM_FRONT",
"CAM_FRONT_RIGHT",
"CAM_FRONT_LEFT",
"CAM_BACK",
"CAM_BACK_LEFT",
"CAM_BACK_RIGHT",
}

# class setting
name_mapping = {
# DBv1.0
"vehicle.car": "car",
"vehicle.construction": "truck",
"vehicle.emergency (ambulance & police)": "car",
"vehicle.motorcycle": "bicycle",
"vehicle.trailer": "trailer",
"vehicle.truck": "truck",
"vehicle.bicycle": "bicycle",
"vehicle.bus (bendy & rigid)": "bus",
"pedestrian.adult": "pedestrian",
"pedestrian.child": "pedestrian",
"pedestrian.construction_worker": "pedestrian",
"pedestrian.personal_mobility": "pedestrian",
"pedestrian.police_officer": "pedestrian",
"pedestrian.stroller": "pedestrian",
"pedestrian.wheelchair": "pedestrian",
"movable_object.barrier": "barrier",
"movable_object.debris": "barrier",
"movable_object.pushable_pullable": "barrier",
"movable_object.trafficcone": "traffic_cone",
"movable_object.traffic_cone": "traffic_cone",
"animal": "animal",
"static_object.bicycle_rack": "bicycle_rack",
# DBv1.1 and UCv2.0
"car": "car",
"truck": "truck",
"bus": "bus",
"trailer": "trailer",
"motorcycle": "bicycle",
"bicycle": "bicycle",
"police_car": "car",
"pedestrian": "pedestrian",
"police_officer": "pedestrian",
"forklift": "car",
"construction_worker": "pedestrian",
"stroller": "pedestrian",
# DBv2.0 and DBv3.0
"animal": "animal",
"movable_object.barrier": "barrier",
"movable_object.pushable_pullable": "barrier",
"movable_object.traffic_cone": "traffic_cone",
"pedestrian.adult": "pedestrian",
"pedestrian.child": "pedestrian",
"pedestrian.construction_worker": "pedestrian",
"pedestrian.personal_mobility": "pedestrian",
"pedestrian.police_officer": "pedestrian",
"pedestrian.stroller": "pedestrian",
"pedestrian.wheelchair": "pedestrian",
"static_object.bicycle rack": "bicycle rack",
"static_object.bollard": "bollard",
"vehicle.ambulance": "car", # Define vehicle.ambulance as car since vehicle.emergency (ambulance & police) is defined as car
"vehicle.bicycle": "bicycle",
"vehicle.bus": "bus",
"vehicle.car": "car",
"vehicle.construction": "truck",
"vehicle.fire": "truck",
"vehicle.motorcycle": "bicycle",
"vehicle.police": "car",
"vehicle.trailer": "trailer",
"vehicle.truck": "truck",
# DBv1.3
"ambulance": "car",
"kart": "car",
"wheelchair": "pedestrian",
"personal_mobility": "pedestrian",
"fire_truck": "truck",
"semi_trailer": "trailer",
"tractor_unit": "truck",
"construction_vehicle": "truck",
"traffic_cone": "traffic_cone",
"trafficcone": "traffic_cone",
"barrier": "barrier",
}

class_names = [
"car",
"truck",
"bus",
"bicycle",
"pedestrian",
"traffic_cone",
"barrier",
]
num_class = len(class_names)
metainfo = dict(classes=class_names)

merge_objects = [
("truck", ["truck", "trailer"]),
]
merge_type = "extend_longer" # One of ["extend_longer","union", None]

# visualization
class_colors = {
"car": (30, 144, 255),
"truck": (140, 0, 255),
"construction_vehicle": (255, 255, 0),
"bus": (111, 255, 111),
"trailer": (0, 255, 255),
"barrier": (0, 0, 0),
"motorcycle": (100, 0, 30),
"bicycle": (255, 0, 30),
"pedestrian": (255, 200, 200),
"traffic_cone": (120, 120, 120),
}
camera_panels = [
"data/CAM_FRONT_LEFT",
"data/CAM_FRONT",
"data/CAM_FRONT_RIGHT",
"data/CAM_BACK_LEFT",
"data/CAM_BACK",
"data/CAM_BACK_RIGHT",
]

filter_attributes = [
("vehicle.bicycle", "vehicle_state.parked"),
("vehicle.bicycle", "cycle_state.without_rider"),
("vehicle.bicycle", "motorcycle_state.without_rider"),
("vehicle.motorcycle", "vehicle_state.parked"),
("vehicle.motorcycle", "cycle_state.without_rider"),
("vehicle.motorcycle", "motorcycle_state.without_rider"),
("bicycle", "vehicle_state.parked"),
("bicycle", "cycle_state.without_rider"),
("bicycle", "motorcycle_state.without_rider"),
("motorcycle", "vehicle_state.parked"),
("motorcycle", "cycle_state.without_rider"),
("motorcycle", "motorcycle_state.without_rider"),
]

evaluator_metric_configs = dict(
evaluation_task="detection",
target_labels=class_names,
center_distance_bev_thresholds=[0.5, 1.0, 2.0, 4.0],
# plane_distance_thresholds is required for the pass fail evaluation
plane_distance_thresholds=[2.0, 4.0],
iou_2d_thresholds=None,
iou_3d_thresholds=None,
label_prefix="autoware",
# bev minimum distance ranges for each range bucket, must be the same length as max_distance,
# they will form bev distance ranges in [(min_distance[0], max_distance[0]), (min_distance[1], max_distance[1]), ...] when filtering
min_distance=[0.0, 50.0, 90.0, 0.0],
# bev maximum distance ranges for each range bucket, must be the same length as min_distance
max_distance=[50.0, 90.0, 121.0, 121.0],
min_point_numbers=0,
matching_class_agnostic_fps=False,
)
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,8 @@
"pedestrian.stroller": "pedestrian",
"pedestrian.wheelchair": "pedestrian",
"movable_object.barrier": "barrier",
"movable_object.debris": "debris",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.debris": "barrier",
"movable_object.pushable_pullable": "barrier",
"movable_object.trafficcone": "traffic_cone",
"movable_object.traffic_cone": "traffic_cone",
"animal": "animal",
Expand All @@ -90,7 +90,7 @@
# DBv2.0 and DBv3.0
"animal": "animal",
"movable_object.barrier": "barrier",
"movable_object.pushable_pullable": "pushable_pullable",
"movable_object.pushable_pullable": "barrier",
"movable_object.traffic_cone": "traffic_cone",
"pedestrian.adult": "pedestrian",
"pedestrian.child": "pedestrian",
Expand Down Expand Up @@ -120,6 +120,9 @@
"semi_trailer": "trailer",
"tractor_unit": "truck",
"construction_vehicle": "truck",
"traffic_cone": "traffic_cone",
"trafficcone": "traffic_cone",
"barrier": "barrier",
}

class_names = [
Expand All @@ -128,7 +131,10 @@
"bus",
"bicycle",
"pedestrian",
"traffic_cone",
"barrier",
]

num_class = len(class_names)
metainfo = dict(classes=class_names)

Expand Down
Loading