我该怎样重写我的DataSet?

img

老哥们,我在搞一个图片分类系统,数据集长这样,我现在得一次读入两张图片作为一个训练样本,我的getItem得一次返回连续的两张图片和他们的一个标签(像这里就是C01),原本的Dataset类已经没法用了,得重写,但是我刚开始学这个python,一直写错,请问我要怎么写我的Dataset类啊?


class MyDataSet(Dataset):
    """customize my dataset"""

    def __init__(self):
        self.images_path = None

    def __len__(self):
        return len(self.images_path)

    def __getitem__(self, item):
        img1 = Image.open(self.images_path[2*item])
        img2 = Image.open(self.images_path[2*item+1])
        label = self.images_class[item] # I don't know how to return the label of 2 images.

        if self.transform is not None:
            img1 = self.transform(img1)
            img2 = self.transform(img2)

        return img1,img2, label

我自己写的半桶水代码长这样,好像也不太能用

琢磨了半天了,很着急,希望老哥们支支招咋重写,感激不尽。

如果你打开图片和获取label的代码没问题的话,直接return img1,img2, label就可以了,不过在调用这个函数的位置返回值要变成3个