python - 在 PyTorch 中设置 root = 时出现无效的语法错误

Question

import torchvision
from torchvision import transforms
train_data_path="./train/"
transforms = transforms.Compose([
transforms.Resize(64),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225] )
])
train_data = torchvision.datasets.ImageFolder
(root=train_data_path,transform=transforms)

这是错误消息：

File "<ipython-input-4-e470172b3902>", line 8
    (root=train_data_path,transform=transforms)
         ^
SyntaxError: invalid syntax

我将如何解决这个问题？

标签： pythonsyntax-errorpytorch

Answer 1

您需要将左括号直接连接到函数，而不需要任何空格分隔它们。尝试将最后两行替换为：

train_data = torchvision.datasets.ImageFolder(
    root=train_data_path, transform=transforms
)