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): ...@@ -365,7 +365,7 @@ def update_nuscenes_infos(pkl_path, out_dir):
print(f'ignore classes: {ignore_class_name}') print(f'ignore classes: {ignore_class_name}')
metainfo = dict() 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: if ignore_class_name:
for ignore_class in ignore_class_name: for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1 metainfo['categories'][ignore_class] = -1
...@@ -505,7 +505,7 @@ def update_kitti_infos(pkl_path, out_dir): ...@@ -505,7 +505,7 @@ def update_kitti_infos(pkl_path, out_dir):
# dataset metainfo # dataset metainfo
metainfo = dict() 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: if ignore_class_name:
for ignore_class in ignore_class_name: for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1 metainfo['categories'][ignore_class] = -1
...@@ -574,7 +574,7 @@ def update_s3dis_infos(pkl_path, out_dir): ...@@ -574,7 +574,7 @@ def update_s3dis_infos(pkl_path, out_dir):
# dataset metainfo # dataset metainfo
metainfo = dict() 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: if ignore_class_name:
for ignore_class in ignore_class_name: for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1 metainfo['categories'][ignore_class] = -1
...@@ -648,7 +648,7 @@ def update_scannet_infos(pkl_path, out_dir): ...@@ -648,7 +648,7 @@ def update_scannet_infos(pkl_path, out_dir):
# dataset metainfo # dataset metainfo
metainfo = dict() 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: if ignore_class_name:
for ignore_class in ignore_class_name: for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1 metainfo['categories'][ignore_class] = -1
...@@ -726,7 +726,7 @@ def update_sunrgbd_infos(pkl_path, out_dir): ...@@ -726,7 +726,7 @@ def update_sunrgbd_infos(pkl_path, out_dir):
# dataset metainfo # dataset metainfo
metainfo = dict() 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: if ignore_class_name:
for ignore_class in ignore_class_name: for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1 metainfo['categories'][ignore_class] = -1
...@@ -835,7 +835,7 @@ def update_lyft_infos(pkl_path, out_dir): ...@@ -835,7 +835,7 @@ def update_lyft_infos(pkl_path, out_dir):
print(f'ignore classes: {ignore_class_name}') print(f'ignore classes: {ignore_class_name}')
metainfo = dict() 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: if ignore_class_name:
for ignore_class in ignore_class_name: for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1 metainfo['categories'][ignore_class] = -1
...@@ -1043,7 +1043,7 @@ def update_waymo_infos(pkl_path, out_dir): ...@@ -1043,7 +1043,7 @@ def update_waymo_infos(pkl_path, out_dir):
# dataset metainfo # dataset metainfo
metainfo = dict() 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: if ignore_class_name:
for ignore_class in ignore_class_name: for ignore_class in ignore_class_name:
metainfo['categories'][ignore_class] = -1 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