import numpy as np import argparse from point_cloud import PointCloud if __name__ == '__main__': parser = argparse.ArgumentParser(description='Write an example point cloud as TXT.') parser.add_argument('output', type=str, help='TXT file for example point cloud.') args = parser.parse_args() point_cloud = PointCloud(np.random.random((100, 3))) point_cloud.to_txt(args.output) print('Wrote %s.' % args.output)
txt_directory = common.dirname(config, 'txt_gt_dir', i) + '%d/' % k utils.makedir(txt_directory) for n in range(N): point_cloud = PointCloud() print('[Data] +') for j in range(perm.shape[0]): txt_file = txt_directories[perm[j]] + '/%d.txt' % n print('[Data] | read ' + txt_file) point_cloud_j = PointCloud.from_txt(txt_file) point_cloud.points = np.concatenate( (point_cloud.points, point_cloud_j.points), axis=0) txt_file = txt_directory + '%d.txt' % n point_cloud.to_txt(txt_file) print('[Data] wrote ' + txt_file) i_inputs = np.swapaxes(i_inputs, 0, 1) i_space = np.swapaxes(i_space, 0, 1) i_sdf_inputs = np.swapaxes(i_sdf_inputs, 0, 1) i_tsdf_inputs = np.swapaxes(i_tsdf_inputs, 0, 1) i_ltsdf_inputs = np.swapaxes(i_ltsdf_inputs, 0, 1) print(i_inputs.shape) print(i_space.shape) print(i_sdf_inputs.shape) print(i_tsdf_inputs.shape) print(i_ltsdf_inputs.shape) utils.write_hdf5(common.filename(config, 'input_file', '.h5', i),