Python 支持向量机在图像分类中的应用

我正在尝试使用支持向量机将计算机上的一组图像分为3类：

我只是面临一个如何加载数据的问题，在下面的示例中，他使用了一个已经保存的数据集。

---

我将所有png格式的图像保存在我电脑的文件夹中

您可以使用枕头将数据加载为numpy阵列，方法如下：

from PIL import Image
import numpy as np

data = np.array(Image.open('yourimg.png')) # .astype(float) if necessary

将其与相结合以读取多个文件，例如

import os
for file in os.listdir('your_dir/'):
    img = Image.open(os.path.join('your_dir/', file))
    data = np.array(img)
    your_model.train(data)

---

但这将只读取一个图像，我想读取文件夹中的所有图像