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

  • Z时代
  • 2024-02-08
  • 分类：IT

我该怎样重写我Dataset里的函数？
大神们，我的数据集长这样，我现在得一次读入两张图片作为一个训练样本，我的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

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

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

以上是 我该怎样重写我Dataset里的函数？ 的全部内容， 来源链接： utcz.com/p/938374.html

回到顶部