循环遍历numpy数组以将其另存为tif文件

Question

我正在尝试使用for循环将形状为（625，256，256，4）的numpy数组（train_images）作为tif图像保存在文件夹中。即，625 x 256像素的RGBN图像。目前，我的代码如下：

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

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

但是不能像这样将numpy数组另存为tif文件。最后，我想在名为0.tif，1.tif等的文件夹中保存625张（RGBN）tif图像。

Answer 1

尝试在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')

希望有帮助。