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

Question

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

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

Answer 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)