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,所以定义的