[エラー] The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

5183 ワード
PyTorch Python error Python テキストリンク

The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

[忙しい人のための解決策]

#後ろにconvert('RGB)をつけてchannels数を3に変更する
input_image = Image.open(filename).convert('RGB')

[原因]　入力された画像のchannel数が4であるために３に戻してあげる必要がある。

<該当コード>

from torchvision import transforms

input_image = Image.open('test.png')  #ここが問題

preprocess = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
    ])
input_tensor = preprocess(input_image)

<エラー画像>

<訂正後のコード>

from torchvision import transforms

input_image = Image.open('test.png')  #ここを訂正した

preprocess = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
    ])
input_tensor = preprocess(input_image)

参考