Skip to content

GitLab

Commit 50ac8884 authored by Xiangxu-0103's avatar Xiangxu-0103 Committed by ZwwWayne
Browse files

fix converter error

parent acdb300d
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
tools/dataset_converters/update_infos_to_v2.py
View file @ 50ac8884
......@@ -365,7 +365,7 @@ def update_nuscenes_infos(pkl_path, out_dir):
print(f'ignore classes: {ignore_class_name}')
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
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
......@@ -505,7 +505,7 @@ def update_kitti_infos(pkl_path, out_dir):
# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
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
......@@ -574,7 +574,7 @@ def update_s3dis_infos(pkl_path, out_dir):
# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
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
......@@ -648,7 +648,7 @@ def update_scannet_infos(pkl_path, out_dir):
# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
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
......@@ -726,7 +726,7 @@ def update_sunrgbd_infos(pkl_path, out_dir):
# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
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
......@@ -835,7 +835,7 @@ def update_lyft_infos(pkl_path, out_dir):
print(f'ignore classes: {ignore_class_name}')
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
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
......@@ -1043,7 +1043,7 @@ def update_waymo_infos(pkl_path, out_dir):
# dataset metainfo
metainfo = dict()
metainfo['categories'] = {k: i for i, k in enumerate(METAINFO['CLASSES'])}
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
......
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