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))
第三方账号登入
QQ 微博 微信