import glob
import os
import numpy as np
from imageio import imsave
import argparse
import sys
sys.path.append('D:/Python/code/ggcnn')
from utils.dataset_processing.image import DepthImage


if __name__ == '__main__':
    path = 'D:/Python/code/cornell'
    pcds = glob.glob(os.path.join(path, 'pcd*[0-9].txt'))
    pcds.sort()

    for pcd in pcds:
        di = DepthImage.from_pcd(pcd, (480, 640))
        di.inpaint()

        of_name = pcd.replace('.txt', 'd.tiff')
        print(of_name)
        imsave(of_name, di.img.astype(np.float32))