这是原始代码
报错:TypeError: ‘builtin_function_or_method’ object is not iterable
这是更改后的代码
报错:TypeError: img should be PIL Image. Got <class ‘torch.Tensor’>
所以我输入tensor是不对的,应该输入PIL image
然而我看网上好多资料说这个函数对tensor和PIL都可以,不知道哪个是对的,我感觉应该是只能输入PIL image不能是tensor
我把输入替换成data_i_pil = transforms.ToPILImage()(data_i)后程序可以正常运行
网上好多说这个函数对tensor和PIL都可以,其实不是的,我举个例子,下面的链接就是其中之一
链接:https://www.jianshu.com/p/ae695df39274