如何为神经网络处理一组图像?

2024-03-28 12:33:19 发布

您现在位置:Python中文网/ 问答频道 /正文

这是我第一次脱离mnist数据集,所以我想知道应该如何处理图像

目前,我有一个装满图像和标签列表的文件夹。我想把图像读入一个数组,然后输入我的CNN

我认为应该这样做:

from PIL import Image
from numpy import array
for i in image_list:
  img = Image.open("i")
  training_set.append(array(img))

我不确定这是否是最有效的方法。如有任何反馈,将不胜感激


Tags: 数据from图像imageimportnumpy文件夹列表
1条回答
网友
1楼 · 发布于 2024-03-28 12:33:19

您可以尝试以下方法来读取图像数据。您可能希望将图像表示为1D数组,但您需要阅读CNN的文档

from PIL import Image
import numpy as np

data = []
for path in image_list:

    # This will be a 3D array
    img = np.asarray(Image.open(path).convert('RGB'))

    # Flatten before appending
    img_flat = img.reshape(img.size)

    data.append(img_flat)

相关问题 更多 >