如何将图像文件夹转换为 npy 文件？答案

【问题标题】：How do I convert a folder of images to a npy file?如何将图像文件夹转换为 npy 文件？
【发布时间】：2022-11-10 11:38:48
【问题描述】：

我有一个包含手势图像的文件夹。但要让它在我的代码上工作，我需要将它更改为 X.npy 和 Y.npy。我寻找了很多关于这类问题的问题，但仍然一无所知。我如何评价这个？如何转换文件夹以创建我自己的 npy 数据集？是否有任何代码或任何转换器？

【问题讨论】：

请澄清您的具体问题或提供其他详细信息以准确突出您的需求。正如它目前所写的那样，很难准确地说出你在问什么。

标签： python image numpy

【解决方案1】：

我在 github 上为此目的找到了一段代码。

from PIL import Image
import os, sys
import cv2
import numpy as np

'''
Converts all images in a directory to '.npy' format.
Use np.save and np.load to save and load the images.
Use it for training your neural networks in ML/DL projects. 
'''

# Path to image directory
path = "/path/to/image/directory/"
dirs = os.listdir( path )
dirs.sort()
x_train=[]

def load_dataset():
    # Append images to a list
    for item in dirs:
        if os.path.isfile(path+item):
            im = Image.open(path+item).convert("RGB")
            im = np.array(im)
            x_train.append(im)

if __name__ == "__main__":
    
    load_dataset()
    
    # Convert and save the list of images in '.npy' format
    imgset=np.array(x_train)
    np.save("imgds.npy",imgset)

【讨论】：

是的，我也看到了，但是我确定运行此程序后是否会得到任何东西。我的意思是在哪里可以找到转换图像的文件夹
我刚刚在collab上尝试了相同的代码。它在根文件夹中创建一个“imgds.npy”文件。自己试试代码。
仍然无法弄清楚，但让我再试一次。不管怎么说，还是要谢谢你

【解决方案2】：

您可以参考以下 GitHub 存储库中的代码 sn-p，我在 google 中找到了将图像文件夹转换为 npy 文件的代码：

https://gist.github.com/anilsathyan7/ffb35601483ac46bd72790fde55f5c04

在这种情况下，文件夹中的所有图像都被转换为 NumPy 数组并附加在名为 x_train 的列表中。要将此图像列表转换并保存在单个“.npy”格式文件中，我们可以使用相同的代码 sn-p ：

    imgset=np.array(x_train)
    np.save("imgds.npy",imgset)

要将此图像列表转换并保存为多个“.npy”格式文件，请使用以下代码 sn-p ：

    imgset=np.array(x_train,dtype=object)
    for i in range(len(imgset)):
        np.save("imgds"+str(i)+".npy",imgset[i])

【讨论】：