Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

[Refactor] update metainfo in pkl file and add categories into metainfo #1934

Merged
merged 5 commits into from
Oct 27, 2022
Merged

[Refactor] update metainfo in pkl file and add categories into metainfo #1934

Changes from 3 commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
d9fdf98
update metainfo in pkl file
ZCMax Oct 20, 2022
bb46d64
update
ZCMax Oct 20, 2022
4e51595
add version for waymo
ZCMax Oct 20, 2022
a563d4b
update kitti metric
ZCMax Oct 25, 2022
ac89f3d
Merge branch 'dev-1.x' of https://github.com/open-mmlab/mmdetection3d…
ZCMax Oct 26, 2022
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
95 changes: 75 additions & 20 deletions tools/dataset_converters/update_infos_to_v2.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -264,10 +264,6 @@ def update_nuscenes_infos(pkl_path, out_dir):
'CLASSES':
('car', 'truck', 'trailer', 'bus', 'construction_vehicle', 'bicycle',
'motorcycle', 'pedestrian', 'traffic_cone', 'barrier'),
'DATASET':
'Nuscenes',
'version':
data_list['metadata']['version']
}
nusc = NuScenes(
version=data_list['metadata']['version'],
Expand Down Expand Up @@ -367,7 +363,16 @@ def update_nuscenes_infos(pkl_path, out_dir):
out_path = osp.join(out_dir, pkl_name)
print(f'Writing to output file: {out_path}.')
print(f'ignore classes: {ignore_class_name}')
converted_data_info = dict(metainfo=METAINFO, data_list=converted_list)

metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
if ignore_class_name:
for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1
metainfo['dataset'] = 'nuscenes'
metainfo['version'] = data_list['metadata']['version']
metainfo['info_version'] = '1.1'
converted_data_info = dict(metainfo=metainfo, data_list=converted_list)

mmengine.dump(converted_data_info, out_path, 'pkl')

Expand Down Expand Up @@ -497,8 +502,16 @@ def update_kitti_infos(pkl_path, out_dir):
out_path = osp.join(out_dir, pkl_name)
print(f'Writing to output file: {out_path}.')
print(f'ignore classes: {ignore_class_name}')
converted_data_info = dict(
metainfo={'DATASET': 'KITTI'}, data_list=converted_list)

# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
if ignore_class_name:
for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1
metainfo['dataset'] = 'kitti'
metainfo['info_version'] = '1.1'
converted_data_info = dict(metainfo=metainfo, data_list=converted_list)

mmengine.dump(converted_data_info, out_path, 'pkl')

Expand Down Expand Up @@ -558,8 +571,17 @@ def update_s3dis_infos(pkl_path, out_dir):
out_path = osp.join(out_dir, pkl_name)
print(f'Writing to output file: {out_path}.')
print(f'ignore classes: {ignore_class_name}')
converted_data_info = dict(
metainfo={'DATASET': 'S3DIS'}, data_list=converted_list)

# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
if ignore_class_name:
for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1
metainfo['dataset'] = 's3dis'
metainfo['info_version'] = '1.1'

converted_data_info = dict(metainfo=metainfo, data_list=converted_list)

mmengine.dump(converted_data_info, out_path, 'pkl')

Expand Down Expand Up @@ -623,8 +645,17 @@ def update_scannet_infos(pkl_path, out_dir):
out_path = osp.join(out_dir, pkl_name)
print(f'Writing to output file: {out_path}.')
print(f'ignore classes: {ignore_class_name}')
converted_data_info = dict(
metainfo={'DATASET': 'SCANNET'}, data_list=converted_list)

# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
if ignore_class_name:
for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1
metainfo['dataset'] = 'scannet'
metainfo['info_version'] = '1.1'

converted_data_info = dict(metainfo=metainfo, data_list=converted_list)

mmengine.dump(converted_data_info, out_path, 'pkl')

Expand Down Expand Up @@ -692,8 +723,17 @@ def update_sunrgbd_infos(pkl_path, out_dir):
out_path = osp.join(out_dir, pkl_name)
print(f'Writing to output file: {out_path}.')
print(f'ignore classes: {ignore_class_name}')
converted_data_info = dict(
metainfo={'DATASET': 'SUNRGBD'}, data_list=converted_list)

# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
if ignore_class_name:
for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1
metainfo['dataset'] = 'sunrgbd'
metainfo['info_version'] = '1.1'

converted_data_info = dict(metainfo=metainfo, data_list=converted_list)

mmengine.dump(converted_data_info, out_path, 'pkl')

Expand All @@ -709,10 +749,6 @@ def update_lyft_infos(pkl_path, out_dir):
'CLASSES':
('car', 'truck', 'bus', 'emergency_vehicle', 'other_vehicle',
'motorcycle', 'bicycle', 'pedestrian', 'animal'),
'DATASET':
'Nuscenes',
'version':
data_list['metadata']['version']
}
print('Start updating:')
converted_list = []
Expand Down Expand Up @@ -797,7 +833,16 @@ def update_lyft_infos(pkl_path, out_dir):
out_path = osp.join(out_dir, pkl_name)
print(f'Writing to output file: {out_path}.')
print(f'ignore classes: {ignore_class_name}')
converted_data_info = dict(metainfo=METAINFO, data_list=converted_list)

metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
if ignore_class_name:
for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1
metainfo['dataset'] = 'lyft'
metainfo['version'] = data_list['metadata']['version']
metainfo['info_version'] = '1.1'
converted_data_info = dict(metainfo=metainfo, data_list=converted_list)

mmengine.dump(converted_data_info, out_path, 'pkl')

Expand Down Expand Up @@ -995,8 +1040,18 @@ def update_waymo_infos(pkl_path, out_dir):
out_path = osp.join(out_dir, pkl_name)
print(f'Writing to output file: {out_path}.')
print(f'ignore classes: {ignore_class_name}')
converted_data_info = dict(
metainfo={'DATASET': 'Waymo'}, data_list=converted_list)

# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
if ignore_class_name:
for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1
metainfo['dataset'] = 'waymo'
metainfo['version'] = '1.2'
metainfo['info_version'] = '1.1'

converted_data_info = dict(metainfo=metainfo, data_list=converted_list)

mmengine.dump(converted_data_info, out_path, 'pkl')

Expand Down