如何将 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)

问题：它只是返回我最后一个类别"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)
        

现在我的目标是以这种方式为我的每个班级格式化我的数据：（点击链接）

format goal