Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
mmdetection3d
Commits
32ab994d
Unverified
Commit
32ab994d
authored
Jan 18, 2023
by
Xiang Xu
Committed by
GitHub
Jan 18, 2023
Browse files
[Fix] Fix cam_list for WaymoDataset (#2187)
parent
5fe12001
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
46 additions
and
49 deletions
+46
-49
configs/_base_/datasets/waymoD5-fov-mono3d-3class.py
configs/_base_/datasets/waymoD5-fov-mono3d-3class.py
+12
-12
configs/_base_/datasets/waymoD5-mv-mono3d-3class.py
configs/_base_/datasets/waymoD5-mv-mono3d-3class.py
+12
-12
configs/_base_/datasets/waymoD5-mv3d-3class.py
configs/_base_/datasets/waymoD5-mv3d-3class.py
+12
-15
mmdet3d/datasets/waymo_dataset.py
mmdet3d/datasets/waymo_dataset.py
+8
-8
tools/dataset_converters/update_infos_to_v2.py
tools/dataset_converters/update_infos_to_v2.py
+2
-2
No files found.
configs/_base_/datasets/waymoD5-fov-mono3d-3class.py
View file @
32ab994d
...
@@ -70,10 +70,10 @@ train_dataloader = dict(
...
@@ -70,10 +70,10 @@ train_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_
RIGH
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_2'
,
CAM_FRONT_
RIGH
T
=
'training/image_2'
,
CAM_SIDE_
RIGH
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_4'
),
CAM_SIDE_
RIGH
T
=
'training/image_4'
),
pipeline
=
train_pipeline
,
pipeline
=
train_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
test_mode
=
False
,
test_mode
=
False
,
...
@@ -97,10 +97,10 @@ val_dataloader = dict(
...
@@ -97,10 +97,10 @@ val_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_
RIGH
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_2'
,
CAM_FRONT_
RIGH
T
=
'training/image_2'
,
CAM_SIDE_
RIGH
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_4'
),
CAM_SIDE_
RIGH
T
=
'training/image_4'
),
ann_file
=
'waymo_infos_val.pkl'
,
ann_file
=
'waymo_infos_val.pkl'
,
pipeline
=
eval_pipeline
,
pipeline
=
eval_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
...
@@ -124,10 +124,10 @@ test_dataloader = dict(
...
@@ -124,10 +124,10 @@ test_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_
RIGH
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_2'
,
CAM_FRONT_
RIGH
T
=
'training/image_2'
,
CAM_SIDE_
RIGH
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_4'
),
CAM_SIDE_
RIGH
T
=
'training/image_4'
),
ann_file
=
'waymo_infos_val.pkl'
,
ann_file
=
'waymo_infos_val.pkl'
,
pipeline
=
eval_pipeline
,
pipeline
=
eval_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
...
...
configs/_base_/datasets/waymoD5-mv-mono3d-3class.py
View file @
32ab994d
...
@@ -70,10 +70,10 @@ train_dataloader = dict(
...
@@ -70,10 +70,10 @@ train_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_
RIGH
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_2'
,
CAM_FRONT_
RIGH
T
=
'training/image_2'
,
CAM_SIDE_
RIGH
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_4'
),
CAM_SIDE_
RIGH
T
=
'training/image_4'
),
pipeline
=
train_pipeline
,
pipeline
=
train_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
test_mode
=
False
,
test_mode
=
False
,
...
@@ -97,10 +97,10 @@ val_dataloader = dict(
...
@@ -97,10 +97,10 @@ val_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_
RIGH
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_2'
,
CAM_FRONT_
RIGH
T
=
'training/image_2'
,
CAM_SIDE_
RIGH
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_4'
),
CAM_SIDE_
RIGH
T
=
'training/image_4'
),
ann_file
=
'waymo_infos_val.pkl'
,
ann_file
=
'waymo_infos_val.pkl'
,
pipeline
=
eval_pipeline
,
pipeline
=
eval_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
...
@@ -124,10 +124,10 @@ test_dataloader = dict(
...
@@ -124,10 +124,10 @@ test_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_
RIGH
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_1'
,
CAM_FRONT_
LEF
T
=
'training/image_2'
,
CAM_FRONT_
RIGH
T
=
'training/image_2'
,
CAM_SIDE_
RIGH
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_3'
,
CAM_SIDE_
LEF
T
=
'training/image_4'
),
CAM_SIDE_
RIGH
T
=
'training/image_4'
),
ann_file
=
'waymo_infos_val.pkl'
,
ann_file
=
'waymo_infos_val.pkl'
,
pipeline
=
eval_pipeline
,
pipeline
=
eval_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
...
...
configs/_base_/datasets/waymoD5-mv3d-3class.py
View file @
32ab994d
...
@@ -76,11 +76,10 @@ train_dataloader = dict(
...
@@ -76,11 +76,10 @@ train_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_RIGHT
=
'training/image_1'
,
CAM_FRONT_LEFT
=
'training/image_1'
,
CAM_FRONT_LEFT
=
'training/image_2'
,
CAM_FRONT_RIGHT
=
'training/image_2'
,
CAM_SIDE_RIGHT
=
'training/image_3'
,
CAM_SIDE_LEFT
=
'training/image_3'
,
CAM_SIDE_LEFT
=
'training/image_4'
,
CAM_SIDE_RIGHT
=
'training/image_4'
),
),
pipeline
=
train_pipeline
,
pipeline
=
train_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
test_mode
=
False
,
test_mode
=
False
,
...
@@ -102,11 +101,10 @@ val_dataloader = dict(
...
@@ -102,11 +101,10 @@ val_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_RIGHT
=
'training/image_1'
,
CAM_FRONT_LEFT
=
'training/image_1'
,
CAM_FRONT_LEFT
=
'training/image_2'
,
CAM_FRONT_RIGHT
=
'training/image_2'
,
CAM_SIDE_RIGHT
=
'training/image_3'
,
CAM_SIDE_LEFT
=
'training/image_3'
,
CAM_SIDE_LEFT
=
'training/image_4'
,
CAM_SIDE_RIGHT
=
'training/image_4'
),
),
pipeline
=
eval_pipeline
,
pipeline
=
eval_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
test_mode
=
True
,
test_mode
=
True
,
...
@@ -127,11 +125,10 @@ test_dataloader = dict(
...
@@ -127,11 +125,10 @@ test_dataloader = dict(
data_prefix
=
dict
(
data_prefix
=
dict
(
pts
=
'training/velodyne'
,
pts
=
'training/velodyne'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT
=
'training/image_0'
,
CAM_FRONT_RIGHT
=
'training/image_1'
,
CAM_FRONT_LEFT
=
'training/image_1'
,
CAM_FRONT_LEFT
=
'training/image_2'
,
CAM_FRONT_RIGHT
=
'training/image_2'
,
CAM_SIDE_RIGHT
=
'training/image_3'
,
CAM_SIDE_LEFT
=
'training/image_3'
,
CAM_SIDE_LEFT
=
'training/image_4'
,
CAM_SIDE_RIGHT
=
'training/image_4'
),
),
pipeline
=
eval_pipeline
,
pipeline
=
eval_pipeline
,
modality
=
input_modality
,
modality
=
input_modality
,
test_mode
=
True
,
test_mode
=
True
,
...
...
mmdet3d/datasets/waymo_dataset.py
View file @
32ab994d
...
@@ -27,10 +27,10 @@ class WaymoDataset(KittiDataset):
...
@@ -27,10 +27,10 @@ class WaymoDataset(KittiDataset):
camera data dict. Defaults to dict(
camera data dict. Defaults to dict(
pts='velodyne',
pts='velodyne',
CAM_FRONT='image_0',
CAM_FRONT='image_0',
CAM_FRONT_
RIGH
T='image_1',
CAM_FRONT_
LEF
T='image_1',
CAM_FRONT_
LEF
T='image_2',
CAM_FRONT_
RIGH
T='image_2',
CAM_SIDE_
RIGH
T='image_3',
CAM_SIDE_
LEF
T='image_3',
CAM_SIDE_
LEF
T='image_4')
CAM_SIDE_
RIGH
T='image_4')
pipeline (List[dict]): Pipeline used for data processing.
pipeline (List[dict]): Pipeline used for data processing.
Defaults to [].
Defaults to [].
modality (dict): Modality to specify the sensor data used
modality (dict): Modality to specify the sensor data used
...
@@ -76,10 +76,10 @@ class WaymoDataset(KittiDataset):
...
@@ -76,10 +76,10 @@ class WaymoDataset(KittiDataset):
data_prefix
:
dict
=
dict
(
data_prefix
:
dict
=
dict
(
pts
=
'velodyne'
,
pts
=
'velodyne'
,
CAM_FRONT
=
'image_0'
,
CAM_FRONT
=
'image_0'
,
CAM_FRONT_
RIGH
T
=
'image_1'
,
CAM_FRONT_
LEF
T
=
'image_1'
,
CAM_FRONT_
LEF
T
=
'image_2'
,
CAM_FRONT_
RIGH
T
=
'image_2'
,
CAM_SIDE_
RIGH
T
=
'image_3'
,
CAM_SIDE_
LEF
T
=
'image_3'
,
CAM_SIDE_
LEF
T
=
'image_4'
),
CAM_SIDE_
RIGH
T
=
'image_4'
),
pipeline
:
List
[
Union
[
dict
,
Callable
]]
=
[],
pipeline
:
List
[
Union
[
dict
,
Callable
]]
=
[],
modality
:
dict
=
dict
(
use_lidar
=
True
),
modality
:
dict
=
dict
(
use_lidar
=
True
),
default_cam_key
:
str
=
'CAM_FRONT'
,
default_cam_key
:
str
=
'CAM_FRONT'
,
...
...
tools/dataset_converters/update_infos_to_v2.py
View file @
32ab994d
...
@@ -859,10 +859,10 @@ def update_waymo_infos(pkl_path, out_dir):
...
@@ -859,10 +859,10 @@ def update_waymo_infos(pkl_path, out_dir):
# pkl generated in the waymo cam only challenage.
# pkl generated in the waymo cam only challenage.
camera_types
=
[
camera_types
=
[
'CAM_FRONT'
,
'CAM_FRONT'
,
'CAM_FRONT_RIGHT'
,
'CAM_FRONT_LEFT'
,
'CAM_FRONT_LEFT'
,
'CAM_
SIDE
_RIGHT'
,
'CAM_
FRONT
_RIGHT'
,
'CAM_SIDE_LEFT'
,
'CAM_SIDE_LEFT'
,
'CAM_SIDE_RIGHT'
,
]
]
print
(
f
'
{
pkl_path
}
will be modified.'
)
print
(
f
'
{
pkl_path
}
will be modified.'
)
if
out_dir
in
pkl_path
:
if
out_dir
in
pkl_path
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment