Зацикливание на массиве NumPy, чтобы сохранить их как TIF - PullRequest
Новая
       9

Зацикливание на массиве NumPy, чтобы сохранить их как TIF

0 голосов
/ 19 ноября 2018

Я пытаюсь сохранить пустой массив (train_images) с формой (625, 256, 256, 4) в виде изображений tif в папке, используя цикл for. То есть 625 изображений RGBN размером 256 x 256 пикселей. В настоящее время мой код выглядит следующим образом: 

path = str(os.getcwd) + "/data/train_images"

for i in train_images:
    num = 0
    i.save(num + '.tif')
    num +=1

Однако невозможно сохранить массив numpy в виде tif-файла, как этот. В конце я хотел бы иметь 625 сохраненных (RGBN) изображений tif в папке с именами 0.tif, 1.tif и т. Д.

1 Ответ

0 голосов
/ 19 ноября 2018

Попробуйте imsave в scipy.misc следующим образом: 

path = str(os.getcwd) + "/data/train_images/"

num = 0
for img in train_images:
    import scipy.misc
    scipy.misc.imsave(path + str(num)  + '.tif', img)
    num +=1

Чтобы прочитать изображение из файла, вам потребуется следующее: 

import imageio
im = imageio.imread('0.tif')

Надеюсь, это поможет.

...