18 lines
747 B
Python
18 lines
747 B
Python
import numpy as np
|
|
import os
|
|
from utils.vis import visualizeUtil
|
|
if __name__ == "__main__":
|
|
root = r"output"
|
|
visualizeUtil.save_all_cam_pos_and_cam_axis(root,"2box", "/home/yan20/nbv_rec/project/franka_control/output/temp")
|
|
# pts_dir_path = "/home/yan20/nbv_rec/project/franka_control/output/2box/pts"
|
|
# pts_dir = os.listdir(pts_dir_path)
|
|
# pts_list = []
|
|
# for i in range(len(pts_dir)):
|
|
# pts_path = os.path.join(pts_dir_path, pts_dir[i])
|
|
# pts = np.load(pts_path)
|
|
# if pts.shape[0] != 0:
|
|
# pts_list.append(pts)
|
|
# combined_pts = np.vstack(pts_list)
|
|
# path = "/home/yan20/nbv_rec/project/franka_control"
|
|
# np.savetxt(os.path.join(path, "combined_pts2.txt"), combined_pts)
|
|
|