hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

merge

Browse Source
This commit is contained in:
hofee 2024-10-19 19:07:40 +08:00
parent d0fbb0f198
commit c9d05f0c86
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
preprocess/preprocessor.py
View File

@ -167,7 +167,7 @@ if __name__ == "__main__":
# scene_list.append(line.strip()) # scene_list.append(line.strip())
scene_list = os.listdir(root) scene_list = os.listdir(root)
from_idx = 0 # 1000 from_idx = 0 # 1000
to_idx = 700 # 1500 to_idx = 1 # 1500
print(scene_list) print(scene_list)

6
utils/vis.py
View File

@ -121,10 +121,10 @@ class visualizeUtil:
# ------ Debug ------ # ------ Debug ------
if __name__ == "__main__": 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" model_dir = r"H:\\AI\\Datasets\\scaled_object_box_meshes"
scene = "omniobject3d-box_030" scene = "cad_model_world"
output_dir = r"C:\Document\Local Project\nbv_rec\nbv_reconstruction\test" 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_cam_pos_and_cam_axis(root, scene, output_dir)
# visualizeUtil.save_all_combined_pts(root, scene, output_dir) # visualizeUtil.save_all_combined_pts(root, scene, output_dir)