AttributeError：'PngImageFile'对象没有属性'shape'

Question

最近，我已经运行了这个项目（https://github.com/milesial/Pytorch-UNet），当我运行Forecast.py时，出现此错误：

Traceback (most recent call last):
  File "predict.py", line 131, in <module>
    device=device)
  File "predict.py", line 43, in predict_img
    transforms.Resize(full_img.shape[2]),
AttributeError: 'PngImageFile' object has no attribute 'shape'

然后我检查代码，发现此版本与之前的版本极为相似

先前版本：

img_height = full_img.size[1]
        tf = transforms.Compose(
            [
                transforms.ToPILImage(),
                transforms.Resize(img_height),
                transforms.ToTensor()
            ]
        )

当前版本：

    tf = transforms.Compose(
        [
            transforms.ToPILImage(),
            transforms.Resize(full_img.shape[1]),
            transforms.ToTensor()
        ]
    )

我不知道发生了什么，但是如果有任何朋友对这个问题有所了解，我将非常感谢您！谢谢！

Answer 1

好像您缺少右圆括号以关闭“撰写”