pytorch怎么样自定义datasets数据集（用于分类任务）一

pytorch怎么样自定义datasets数据集（用于分类任务），做个实验 ，如下的结构。其中voc30里面存放这三十张图片，图片名为000001.jpg到0000030.jpg

而作为标签文件的labels2.txt存放的是上述图片的地址和类别标签，如下

from torchvision import transforms, utils
from torch.utils.data import Dataset, DataLoader
import matplotlib.pyplot as plt
from PIL import Image

def default_loader(path):
return Image.open(path).convert(‘RGB’)

from torchvision import transforms, utils
from torch.utils.data import Dataset, DataLoader
import matplotlib.pyplot as plt
from PIL import Image

def default_loader(path):
    return Image.open(path).convert('RGB')

class MyDataset(Dataset)