Skip to content

GitLab

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

[Fix] Fix sunrgbd converter (#2440)

parent bc849cc9
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/create_data.py
View file @ a65171ab
......@@ -162,8 +162,8 @@ def sunrgbd_data_prep(root_path, info_prefix, out_dir, workers):
root_path, info_prefix, out_dir, workers=workers)
info_train_path = osp.join(out_dir, f'{info_prefix}_infos_train.pkl')
info_val_path = osp.join(out_dir, f'{info_prefix}_infos_val.pkl')
update_pkl_infos('scannet', out_dir=out_dir, pkl_path=info_train_path)
update_pkl_infos('scannet', out_dir=out_dir, pkl_path=info_val_path)
update_pkl_infos('sunrgbd', out_dir=out_dir, pkl_path=info_train_path)
update_pkl_infos('sunrgbd', out_dir=out_dir, pkl_path=info_val_path)
def waymo_data_prep(root_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