ARTeLab/DTM_Estimation_SRandD

1from osgeo import gdal2import os3from PIL import Image4import numpy as np5 6path = '/home/super/datasets-nas/hirise_oxia_planum_test_tiles_thruth/'7 8for i, file_name in enumerate(os.listdir(path)[40:90]):9    file_path = os.path.join(path, file_name)10    x = gdal.Open(file_path)11    x_array = x.ReadAsArray()12    # print(x_array.shape)13    pil_img = Image.fromarray(np.uint8(x_array), 'L')14    pil_img.save(f'demo_imgs/{i}.png')15