python - 使用 sklearn load_files() 从 png 加载图像作为数据

Question

我目前正在使用 sklearn 构建一个简单的图像识别器。

我需要使用 load_files('./directory/') 从该目录中的子文件夹加载图像。

它正确获取了目标值，但数据属性不是简单的像素值。我假设我需要设置编码参数来考虑图像文件，但找不到确切的用途。

Answer 1

编码参数用于解码假定文本编码(例如 UTF-8)的文件内容的原始字节。

对于图像文件，您需要自己迭代filenames 属性的内容并使用类似scipy.misc.imread 的内容。 (您还需要安装 PIL 或 Pillow 包)。

这是一个实用函数，用于从野外标记的人脸中加载 jpeg 文件的数据作为 numpy 数组:

https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/datasets/lfw.py#L108

您可以使用它来了解如何编写您自己的自定义数据集加载器。