Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
69 commits
Select commit Hold shift + click to select a range
4b90876
Add pretrain datasets
Jan 5, 2026
0d55e38
Add
Jan 5, 2026
7a5a018
Add pretrain datasets
Jan 5, 2026
c22672c
Add
Jan 6, 2026
1f69a71
Add
Jan 6, 2026
bfcda39
Add
Jan 6, 2026
84b804b
Add
Jan 8, 2026
40d103f
Update configs
KSeangTan Jan 10, 2026
82f5e92
Update configs
KSeangTan Jan 12, 2026
8f50e47
Update config delimiter
KSeangTan Jan 12, 2026
d428783
Update config delimiter
KSeangTan Jan 12, 2026
9bb04ce
Update docker installation commit for perception_evaluation
KSeangTan Jan 14, 2026
ebee4a0
Merge branch 'tier4:main' into main
KSeangTan Jan 14, 2026
0c640a6
Add location and vehicle type to prefix frame
KSeangTan Jan 15, 2026
1657df2
Add polars to installation
KSeangTan Jan 20, 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
f237640
Adjust configs
KSeangTan Jan 23, 2026
05c028c
Add configs
KSeangTan Jan 23, 2026
c07348a
Update comments
KSeangTan Jan 23, 2026
6ec0b80
Update comments
KSeangTan Jan 23, 2026
ab8d836
Update configs
KSeangTan Jan 23, 2026
71e082f
Update comments
KSeangTan Jan 24, 2026
2b25540
Update comments
KSeangTan Jan 25, 2026
a3803e6
Update creation of infos
KSeangTan Jan 26, 2026
f443431
Update creation of infos
KSeangTan Jan 26, 2026
c5f356a
Update creation of infos
KSeangTan Jan 28, 2026
10d8761
Fix img roi
KSeangTan Jan 28, 2026
f44c169
Fix t4datsets
KSeangTan Jan 28, 2026
c2159f8
Fix t4datsets
KSeangTan Jan 28, 2026
a005bfe
Fix versions
KSeangTan Jan 28, 2026
242d0ff
Fix versions
KSeangTan Jan 28, 2026
f844876
Fix versions
KSeangTan Jan 29, 2026
4603f0c
Fix versions
KSeangTan Jan 29, 2026
93874f9
Update configs
KSeangTan Jan 29, 2026
63ff81f
ci(pre-commit): autofix
pre-commit-ci[bot] Jan 29, 2026
7da5fe7
Update configs
KSeangTan Jan 29, 2026
6abce1e
Merge branch 'feat/train_centerpoint_2_6' of github.com:KSeangTan/AWM…
KSeangTan Jan 29, 2026
cab45ba
ci(pre-commit): autofix
pre-commit-ci[bot] Jan 29, 2026
db3666a
Update configs
KSeangTan Jan 29, 2026
56b28fb
Merge branch 'feat/train_centerpoint_2_6' of github.com:KSeangTan/AWM…
KSeangTan Jan 29, 2026
152e23b
Update configs
KSeangTan Jan 30, 2026
13cfcee
Added
KSeangTan Jan 30, 2026
36940b2
ci(pre-commit): autofix
pre-commit-ci[bot] Jan 30, 2026
6dcc673
Update configs
KSeangTan Jan 30, 2026
54c8b15
Update configs
KSeangTan Jan 30, 2026
1d073df
ci(pre-commit): autofix
pre-commit-ci[bot] Jan 30, 2026
73ea06b
Update configs
KSeangTan Feb 1, 2026
eb456ec
Merge branch 'feat/train_centerpoint_2_6' of github.com:KSeangTan/AWM…
KSeangTan Feb 1, 2026
5687647
Update configs
KSeangTan Feb 1, 2026
76d7e89
ci(pre-commit): autofix
pre-commit-ci[bot] Feb 1, 2026
ef81afd
Update configs
KSeangTan Feb 1, 2026
89acaa6
Update configs
KSeangTan Feb 1, 2026
4da86aa
Merge branch 'feat/train_centerpoint_2_6' of github.com:KSeangTan/AWM…
KSeangTan Feb 1, 2026
52f19dd
Update configs
KSeangTan Feb 1, 2026
a4dfc80
Update configs
KSeangTan Feb 2, 2026
90f45b8
Update configs
KSeangTan Feb 7, 2026
01590d1
Update configs
KSeangTan Feb 7, 2026
4bc972f
Update configs
KSeangTan Feb 7, 2026
2947208
Update configs
KSeangTan Feb 9, 2026
b2ee41d
Update configs
KSeangTan Feb 10, 2026
b7465a5
Update configs
KSeangTan Feb 13, 2026
e5af033
Update configs
KSeangTan Feb 13, 2026
affc7f8
Update configs
KSeangTan Feb 14, 2026
c5a611c
Update configs
KSeangTan Feb 14, 2026
fef6216
Update configs
KSeangTan Feb 14, 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
2 changes: 2 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,8 @@ RUN python3 -m pip install git+https://github.com/tier4/autoware_perception_eval
RUN python3 -m pip --no-cache-dir install \
setuptools==60.2.0 \
transformers==4.51.3 \
onnx_graphsurgeon==0.5.8 \
spconv-cu120==2.3.6 \
polars==1.37.1

# NOTE(knzo25): this patch is needed to use numpy versions over 1.23.5 (version used in mmdet3d 1.4.0)
Expand Down
18 changes: 11 additions & 7 deletions autoware_ml/configs/detection3d/dataset/t4dataset/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,8 +28,10 @@
"db_j6gen2_v6",
"db_j6gen2_v7",
"db_j6gen2_v8",
"db_j6gen2_v9",
"db_largebus_v1",
"db_largebus_v2",
"db_largebus_v3",
"db_jpntaxi_v1",
"db_jpntaxi_v2",
"db_jpntaxi_v4",
Expand All @@ -39,14 +41,16 @@
"db_j6_v3",
"db_j6_v5",
]

# TODO (KokSeang): This will be removed to avoid repeatitive computation
# Dataset set, test info files, and enable/disable evaluation of prefix
dataset_test_groups = {
"db_jpntaxi_gen2_base": "t4dataset_jpntaxi_gen2_base_infos_test.pkl",
"db_j6gen2": "t4dataset_j6gen2_infos_test.pkl",
"db_j6gen2_v6": "t4dataset_j6gen2_v6_infos_test.pkl",
"db_largebus": "t4dataset_largebus_infos_test.pkl",
"db_jpntaxi": "t4dataset_xx1_infos_test.pkl",
"db_j6": "t4dataset_x2_infos_test.pkl",
"db_base": "t4dataset_base_infos_test.pkl",
"j6gen2_base": ("t4dataset_j6gen2_base_infos_test.pkl", False),
"j6gen2": ("t4dataset_j6gen2_infos_test.pkl", False),
"largebus": ("t4dataset_largebus_infos_test.pkl", False),
"jpntaxi_base": ("t4dataset_jpntaxi_base_infos_test.pkl", False),
"jpntaxi_gen2": ("t4dataset_jpntaxi_gen2_infos_test.pkl", False),
"base": ("t4dataset_base_infos_test.pkl", True),
}

# dataset format setting
Expand Down
6 changes: 2 additions & 4 deletions autoware_ml/configs/detection3d/dataset/t4dataset/j6gen2.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,13 +26,11 @@
"db_j6gen2_v6",
"db_j6gen2_v7",
"db_j6gen2_v8",
"db_j6gen2_v9",
]

dataset_test_groups = {
"db_j6gen2_v6": "t4dataset_j6gen2_v6_infos_test.pkl",
"db_j6gen2": "t4dataset_j6gen2_infos_test.pkl",
"db_largebus": "t4dataset_largebus_infos_test.pkl",
"db_gen2_base": "t4dataset_gen2_base_infos_test.pkl",
"j6gen2": ("t4dataset_j6gen2_infos_test.pkl", True),
}

# dataset format setting
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,15 +26,16 @@
"db_j6gen2_v6",
"db_j6gen2_v7",
"db_j6gen2_v8",
"db_j6gen2_v9",
"db_largebus_v1",
"db_largebus_v2",
"db_largebus_v3",
]

dataset_test_groups = {
"db_j6_gen2_v6": "t4dataset_j6gen2_v6_infos_test.pkl",
"db_j6gen2": "t4dataset_j6gen2_infos_test.pkl",
"db_largebus": "t4dataset_largebus_infos_test.pkl",
"db_j6_gen2_base": "t4dataset_j6gen2_base_infos_test.pkl",
"largebus": ("t4dataset_largebus_infos_test.pkl", False),
"j6gen2": ("t4dataset_j6gen2_infos_test.pkl", False),
"j6gen2_base": ("t4dataset_j6gen2_base_infos_test.pkl", True),
}

# dataset format setting
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,8 @@

# dataset scene setting
dataset_test_groups = {
"db_jpntaxi_gen2_base": "t4dataset_jpntaxi_gen2_base_infos_test.pkl",
"db_jpntaxi_gen2_v2": "t4dataset_jpntaxi_gen2_v2_infos_test.pkl",
"db_jpntaxi_base": "t4dataset_jpntaxi_base_infos_test.pkl",
"jpntaxi_gen2": ("t4dataset_jpntaxi_gen2_infos_test.pkl", False),
"jpntaxi_base": ("t4dataset_jpntaxi_base_infos_test.pkl", True),
}

dataset_version_list = [
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,23 +8,24 @@

# dataset type setting
dataset_type = "T4Dataset"
info_train_file_name = "t4dataset_j6gen2_v6_infos_train.pkl"
info_val_file_name = "t4dataset_j6gen2_v6_infos_val.pkl"
info_test_file_name = "t4dataset_j6gen2_v6_infos_test.pkl"
info_train_file_name = "t4dataset_jpntaxi_gen2_infos_train.pkl"
info_val_file_name = "t4dataset_jpntaxi_gen2_infos_val.pkl"
info_test_file_name = "t4dataset_jpntaxi_gen2_infos_test.pkl"

info_train_statistics_file_name = "t4dataset_j6gen2_v6_statistics_train.parquet"
info_val_statistics_file_name = "t4dataset_j6gen2_v6_statistics_val.parquet"
info_test_statistics_file_name = "t4dataset_j6gen2_v6_statistics_test.parquet"
info_train_statistics_file_name = "t4dataset_jpntaxi_gen2_statistics_train.parquet"
info_val_statistics_file_name = "t4dataset_jpntaxi_gen2_statistics_val.parquet"
info_test_statistics_file_name = "t4dataset_jpntaxi_gen2_statistics_test.parquet"

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

dataset_test_groups = {
"db_j6gen2_v6": "t4dataset_j6gen2_v6_infos_test.pkl",
"jpntaxi_gen2": ("t4dataset_jpntaxi_gen2_infos_test.pkl", True),
}

dataset_version_list = [
"db_jpntaxigen2_v1",
"db_jpntaxigen2_v2",
]

# dataset format setting
data_prefix = dict(
pts="",
Expand All @@ -36,7 +37,6 @@
CAM_BACK_LEFT="",
sweeps="",
)

camera_types = {
"CAM_FRONT",
"CAM_FRONT_RIGHT",
Expand Down Expand Up @@ -156,6 +156,7 @@
"data/CAM_BACK_RIGHT",
]

# Add filter attributes
filter_attributes = [
("vehicle.bicycle", "vehicle_state.parked"),
("vehicle.bicycle", "cycle_state.without_rider"),
Expand Down
192 changes: 0 additions & 192 deletions autoware_ml/configs/detection3d/dataset/t4dataset/jpntaxi_gen2_base.py

This file was deleted.

Loading