TypeError: object of type ‘Living_Body‘ has no len()

此篇博客讲述了在使用 PyTorch 数据加载时遇到 Living_Body 类型错误,通过在Dataset类中添加 `__len__` 方法计算数据量并修复了类型问题。作者分享了代码片段和解决步骤,适合对数据处理和PyTorch数据加载有疑问的读者。
摘要由CSDN通过智能技术生成

报错:

Traceback (most recent call last):
  File "train_unsupervised.py", line 107, in <module>
    drop_last=False)
  File "/home/xinying/self-supervised-relational-reasoning-master/datamanager.py", line 417, in get_train_loader
    num_workers=num_workers, pin_memory=True, drop_last=drop_last)
  File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 224, in __init__
    sampler = RandomSampler(dataset, generator=generator)
  File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/sampler.py", line 94, in __init__
    if not isinstance(self.num_samples, int) or self.num_samples <= 0:
  File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/sampler.py", line 102, in num_samples
    return len(self.data_source)
TypeError: object of type 'Living_Body' has no len()

在数据加载中,加入len函数,用于计算总的数据量是多少

class Living_Body(Dataset):
    def __init__(self, img_path, repeat_augmentations, **kwds):
        super().__init__()
        self.repeat_augmentations = repeat_augmentations
        self.img_path = img_path
        self.imgs = []
        self.targets = []
        
        for img in os.listdir(self.img_path):
            img_path = os.path.join(self.img_path, img)
            self.imgs.append(img_path)
            
    def __len__(self):
        return len(self.imgs)
            
    def __getitem__(self, index):
        print(self.data[index])
        #img, target = self.data[index], self.targets[index]
        img = Image.open(self.imgs[index])
        
        pic = Image.fromarray(img)
            
        img_list = list()
        if self.transform is not None:
            for _ in range(self.repeat_augmentations):
               img_transformed = self.transform(pic.copy())
               img_list.append(img_transformed)
        else:
            img_list = None
           
        if self.target_transform is not None:
            target = self.target_transform(target)

        return img, img_list, img#return img, img_list, target
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

我现在强的可怕~

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值