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

Question

通过pytorch训练resnet50模型我得到了这个错误：

RuntimeError：无效参数0：张量的大小必须匹配，除了维度0.在/pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333中，维度3中的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')