pytorch制作数据集

在数据的存储方式是同一个类的图片都放在同一个文件夹，且该文件夹被命名为该类标签的名称时，在该种前提下制作数据集。
所用的是继承pytorch中Dataset类，并重写该类中的init函数和getitem函数。代码如下：

from torch.utils.data import Dataset
from PIL import Image
import os


class MyData(Dataset):

    # 类的初始化，用于输入存放图片的根目录文件夹和存放图片的文件夹（此例中存放图片的文件夹名即为文件夹中所有图片的类型），给自身属性赋值
    def __init__(self