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)