From c9d05f0c867af968682ef0fbef117ee4adbf7603 Mon Sep 17 00:00:00 2001 From: hofee Date: Sat, 19 Oct 2024 19:07:40 +0800 Subject: [PATCH] merge --- preprocess/preprocessor.py | 2 +- utils/vis.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/preprocess/preprocessor.py b/preprocess/preprocessor.py index c6a9338..ad8b166 100644 --- a/preprocess/preprocessor.py +++ b/preprocess/preprocessor.py @@ -167,7 +167,7 @@ if __name__ == "__main__": # scene_list.append(line.strip()) scene_list = os.listdir(root) from_idx = 0 # 1000 - to_idx = 700 # 1500 + to_idx = 1 # 1500 print(scene_list) diff --git a/utils/vis.py b/utils/vis.py index 78992e2..5f61a6d 100644 --- a/utils/vis.py +++ b/utils/vis.py @@ -121,10 +121,10 @@ class visualizeUtil: # ------ Debug ------ if __name__ == "__main__": - root = r"C:\Document\Local Project\nbv_rec\nbv_reconstruction\temp" + root = r"/home/yan20/nbv_rec/project/franka_control/temp" model_dir = r"H:\\AI\\Datasets\\scaled_object_box_meshes" - scene = "omniobject3d-box_030" - output_dir = r"C:\Document\Local Project\nbv_rec\nbv_reconstruction\test" + scene = "cad_model_world" + output_dir = r"/home/yan20/nbv_rec/project/franka_control/temp/output" # visualizeUtil.save_all_cam_pos_and_cam_axis(root, scene, output_dir) # visualizeUtil.save_all_combined_pts(root, scene, output_dir)