--------------------------------------------------------------------------- TypeError Traceback (most recent call last) Cell In[31], line 8 6 print('sample:',x.shape,y.shape) 7 if __name__ == '__main__': ----> 8 main() Cell In[31], line 5, in main() 2 viz= visdom.Visdom() 4 db = Pokemon('pokemon',224,'train') #因为在当前文件夹下只需要文件名 ----> 5 x,y = next(iter(db)) 6 print('sample:',x.shape,y.shape) Cell In[30], line 87, in Pokemon.__getitem__(self, idx) 76 img, label = self.images[idx], self.labels[idx] 77 tf = transforms.Compose([ 78 lambda x:Image.open(img).convert('RGB'), #string path=>image data 79 transforms.Resize((self.resize,self.resize)), (...) 85 """ 86 ]) ---> 87 img = tf(img) 88 label = torch.tensor(label) 89 return img,label File D:\Anaconda\envs\gpu_pytorch\lib\site-packages\torchvision\transforms\transforms.py:95, in Compose.__call__(self, img) 93 def __call__(self, img): 94 for t in self.transforms: ---> 95 img = t(img) 96 return img TypeError: 'str' object is not callable
时间: 2023-09-25 16:12:07 浏览: 105
这个错误是由于在你的代码中使用了一个字符串对象作为函数调用。具体而言,在你的代码的第87行,你尝试将一个字符串路径传递给`transforms.Compose`,但是它期望接收一个可调用的对象。
要解决这个问题,你需要确保你传递给`transforms.Compose`的是一个可调用的对象,而不是一个字符串。你可以使用`transforms.ToPILImage()`函数将路径转换为图像数据对象,然后再将其传递给`transforms.Compose`。以下是修改后的代码示例:
```python
from PIL import Image
import torchvision.transforms as transforms
tf = transforms.Compose([
transforms.ToPILImage(),
lambda x: x.convert('RGB'),
transforms.Resize((self.resize, self.resize)),
# 其他转换操作
])
img = tf(img)
```
通过这样的修改,你应该能够解决这个错误并继续运行你的代码。
阅读全文