From 9159f563759ce3547fc6e1f4d15f439f6a7bafc1 Mon Sep 17 00:00:00 2001 From: Xiang Xu Date: Tue, 18 Apr 2023 19:53:33 +0800 Subject: [PATCH] fix sunrgbd converter --- tools/create_data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/create_data.py b/tools/create_data.py index 05fd98cf8..f3c85fce6 100644 --- a/tools/create_data.py +++ b/tools/create_data.py @@ -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,