使用pytorch中的transform提示 'ToTensor' object is not iterable

刚刚在Pytorch中对数据进行处理的时候,我写了下面这一行代码:

 train_folder_set = ImageFolder(filename + '/train', transform=transforms.Compose(
     transforms.ToTensor(),
    ))

然后使用一个循环输出:

for i in train_folder_set:
	pint(i)

运行,然后就报错了:

TypeError: 'ToTensor' object is not iterable

按照道理来说,不应该啊,后来才发现,是transform.Compose()使用的时候要加[],然后将代码改成下面这张:

    train_folder_set = ImageFolder(filename + '/train', transform=transforms.Compose([
     transforms.ToTensor(),
    ]))

问题就解决了,所以现在记录一下,为可能遇到相同问题的同学提供思路。

2020 4.21

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值