nbv_rec_control/combine_all_pts.py
2024-10-13 19:47:05 +08:00

15 lines
534 B
Python

import numpy as np
import os
if __name__ == "__main__":
pts_dir_path = "/home/yan20/nbv_rec/project/franka_control/temp_output/cad_model_world/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.loadtxt(pts_path)
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_pts.txt"), combined_pts)