深度学习-制作自己的数据集

1.背景

对于文件夹套子文件夹，子文件夹里有很多同一类别的图像，子文件夹表示里面图像的类别。这种结构的文件叫做图像层叠式即为imageFolder。如下所示：

该文件夹下面有10个类别的子文件夹，如下：

 

每个子文件夹下有许多图像，如下：

 2.开始制作数据集

"""
    对于一个文件夹下有多个文件夹，每个子文件夹都是子文件夹里图片的类别名，这种称谓图像层叠形式imageFolder。
"""
import os
from torch.utils.data import Dataset, DataLoader, random_split
import pandas as pd
import numpy as np


dataset_dir = "F:/reshape_images"

# 获取子文件夹的个数，也就是类别数
classes = os.listdir(dataset_dir)
print(classes)

# 定义一个数组存放filename, filepath, label,所以定义的