如何在 scikit-learn 中加载图像数据集？答案

【问题标题】：How to load an image dataset in scikit-learn?如何在 scikit-learn 中加载图像数据集？
【发布时间】：2020-09-24 17:44:31
【问题描述】：

我收集了一组图像，我想在这些图像上训练模型。

如何加载图像数据集？我有一个训练数据文件夹，其中有两个文件夹，表示两种不同类型的对象。我将如何加载此数据集然后训练模型？

【问题讨论】：

欢迎堆栈溢出。不幸的是，这不是代码编写或教程服务，我们要求问题包括 minimal reproducible example，包括 code，以了解您迄今为止根据自己的研究尝试过的内容，以及出现了什么问题您的尝试
@G.Anderson 你说的“不幸”是什么意思？
@desertnaut 我的意思是，不幸的是，对于 OP，他们无法获得他们请求的特定类型的帮助。这是向新网站用户介绍这一点的一种礼貌方式，试图在欢迎他们的同时仍然告诉他们需要重新表述问题

标签： python machine-learning scikit-learn

【解决方案1】：

这可能会帮助您将数据集从单个图像文件夹加载到 data 变量中

import cv2
import os
import numpy as np
path = 'path to your dataset'
list_of_files = os.listdir(path)
data = np.empty(0)
for i in list_of_files:
    x = cv2.imread(os.path.join(path+i))
    data.append(x)

【讨论】：