如何将 tiff 图像保存到新的 npy 文件中？

Question

我想将一些 tiff 图像保存到新的 npy 文件中。

我的数据保存在 5 个不同的文件中（tiff 格式）。我想访问它们中的每一个，将它们转换为 narray，然后将它们保存在一个新的 npy 文件中（用于深度学习分类）。

import numpy as np 
from PIL import Image
import os

Data_dir = r"C:\Desktop\Université_2019_2020\CoursS2_Mosef\Stage\Data\Grand_Leez\shp\imagettes"
Categories = ["Bouleau_tiff", "Chene_tiff", "Erable_tiff", "Frene_tiff", "Peuplier_tiff"]

for categorie in Categories:
    path = os.path.join(Data_dir, categorie) #path for each species
    for img in os.listdir(path):
        path_img = os.path.join(path,img)
        im = Image.open(os.path.join(path_img)) #load an image file
        imarray = np.array(im) # convert it to a matrix
        imarray = np.delete(imarray, 3, axis=2)
        np.save(Data_dir, imarray)

问题：只是return我上次观察的最后一类"Peuplier_tiff"，还存入了名字imagette，不知道为什么。

最后但同样重要的是，我对我的目标有疑问，我如何才能确保我的类别正确分配给相应的数组。

很多问题，预先感谢您的帮助。

S.V

Answer 1

感谢您的回复。它使用此代码：

import numpy as np
from PIL import Image
import os


new_dir = "dta_npy"
directory = r"C:\Desktop\Université_2019_2020\CoursS2_Mosef\Stage\Data\Grand_Leez\shp\imagettes"
Data_dir = os.path.join(directory, new_dir)
os.makedirs(Data_dir)
print("Directory '%s' created" %Data_dir)

Categories = ["Bouleau_tif","Chene_tif", "Erable_tif", "Frene_tif", "Peuplier_tif"]

for categorie in Categories:
    path = os.path.join(directory,categorie) #path for each species
    for img in os.listdir(path):
        im = Image.open(os.path.join(path,img)) #load an image file
        imarray = np.array(im) # convert it to a matrix
        imarray = np.delete(imarray, 3, axis=2)
        unique_name=img
        unique_name = unique_name.split(".")
        unique_name = unique_name[0]
        np.save(Data_dir+"/"+unique_name, imarray)

现在我的 objective 是格式化我的数据，对于我的每个 class，这样：（点击 link）

format goal

如何将 tiff 图像保存到新的 npy 文件中？

How to save tiff images into a new npy file?

python

tiff

image

save

multidimensional-array