File size: 429 Bytes
4216279
 
 
 
 
 
 
b80f9db
4216279
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from osgeo import gdal
import os
from PIL import Image
import numpy as np

path = '/home/super/datasets-nas/hirise_oxia_planum_test_tiles_thruth/'

for i, file_name in enumerate(os.listdir(path)[40:90]):
    file_path = os.path.join(path, file_name)
    x = gdal.Open(file_path)
    x_array = x.ReadAsArray()
    # print(x_array.shape)
    pil_img = Image.fromarray(np.uint8(x_array), 'L')
    pil_img.save(f'demo_imgs/{i}.png')