RuntimeError: неверный аргумент 0: размеры тензоров должны совпадать, за исключением измерения 0. Получены 3 и 1 в измерении 1

Question

При обучении модели resnet50 через pytorch я получил эту ошибку:

RuntimeError: неверный аргумент 0: размеры тензоров должны совпадать, кроме как в измерении 0. Получил 3 и 1 в измерении 1 в / 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')