python - RuntimeError：无效参数 0：张量的大小必须匹配，但维度 0 除外。得到维度 1 中的 3 和 1

Question

在通过 pytorch 训练 resnet50 模型时出现此错误：

RuntimeError：无效参数 0：张量的大小必须匹配，但维度 0 除外。在 /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333 的维度 1 中获得 3 和 1

我正在使用这个：http: //github.com/Helias/Car-Model-Recognition/

使用此数据集http://vmmrdb.cecsresearch.org/

Answer 1

我解决了这个问题，问题是图像颜色通道不同，并非所有图像都是RGB，所以我在dataset.py中进行了转换，我改变了这个：

im = Image.open(image_path)

进入这个：

im = Image.open(image_path).convert('RGB')