我该怎样重写我Dataset里的函数?

新手上路,请多包涵

image.png
大神们,我的数据集长这样,我现在得一次读入两张图片作为一个训练样本,我的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

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

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

阅读 741
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题