Skip to content

GitLab

Unverified Commit 273c7145 authored by Sun Jiahao's avatar Sun Jiahao Committed by GitHub
Browse files

[Fix]: Fix scannet info update (#2332) 

* fix polarmix UT

* fix scannet info bug
parent 3e965912
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/dataset_converters/update_infos_to_v2.py
View file @ 273c7145
...@@ -630,6 +630,7 @@ def update_scannet_infos(pkl_path, out_dir): ...@@ -630,6 +630,7 @@ def update_scannet_infos(pkl_path, out_dir):
# TODO support camera # TODO support camera
# np.linalg.inv(info['axis_align_matrix'] @ extrinsic): depth2cam # np.linalg.inv(info['axis_align_matrix'] @ extrinsic): depth2cam
anns = ori_info_dict.get('annos', None) anns = ori_info_dict.get('annos', None)
ignore_class_name = set()
if anns is not None: if anns is not None:
temp_data_info['axis_align_matrix'] = anns[ temp_data_info['axis_align_matrix'] = anns[
'axis_align_matrix'].tolist() 'axis_align_matrix'].tolist()
...@@ -637,7 +638,6 @@ def update_scannet_infos(pkl_path, out_dir): ...@@ -637,7 +638,6 @@ def update_scannet_infos(pkl_path, out_dir):
instance_list = [] instance_list = []
else: else:
num_instances = len(anns['name']) num_instances = len(anns['name'])
ignore_class_name = set()
instance_list = [] instance_list = []
for instance_id in range(num_instances): for instance_id in range(num_instances):
empty_instance = get_empty_instance() empty_instance = get_empty_instance()
......
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