Question

我正在尝试使用SVM将我的计算机上的图像分为3类：

我正面临如何加载数据的问题，如下例所示，他使用的是已保存的数据集。 http://scikit-learn.org/stable/auto_examples/classification/plot_digits_classification.html

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

Answer 1

您可以使用Pillow将数据作为numpy数组加载，这样：

from PIL import Image
import numpy as np

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

将其与os.listdir结合使用以读取多个文件，例如

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)

支持矢量机器，用于分类图像

1 个答案: