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)
Esempio n. 2
0
                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),