RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1

Question

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

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333

我正在使用这个：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')