Unverified Commit 32ab994d authored by Xiang Xu's avatar Xiang Xu Committed by GitHub
Browse files

[Fix] Fix cam_list for WaymoDataset (#2187)

parent 5fe12001
...@@ -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_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,
...@@ -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_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'),
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_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'),
ann_file='waymo_infos_val.pkl', ann_file='waymo_infos_val.pkl',
pipeline=eval_pipeline, pipeline=eval_pipeline,
modality=input_modality, modality=input_modality,
......
...@@ -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_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,
...@@ -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_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'),
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_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'),
ann_file='waymo_infos_val.pkl', ann_file='waymo_infos_val.pkl',
pipeline=eval_pipeline, pipeline=eval_pipeline,
modality=input_modality, modality=input_modality,
......
...@@ -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,
......
...@@ -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_RIGHT='image_1', CAM_FRONT_LEFT='image_1',
CAM_FRONT_LEFT='image_2', CAM_FRONT_RIGHT='image_2',
CAM_SIDE_RIGHT='image_3', CAM_SIDE_LEFT='image_3',
CAM_SIDE_LEFT='image_4') CAM_SIDE_RIGHT='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_RIGHT='image_1', CAM_FRONT_LEFT='image_1',
CAM_FRONT_LEFT='image_2', CAM_FRONT_RIGHT='image_2',
CAM_SIDE_RIGHT='image_3', CAM_SIDE_LEFT='image_3',
CAM_SIDE_LEFT='image_4'), CAM_SIDE_RIGHT='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',
......
...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment