torchvision.utils.make_grid は画像チャンネルを変更します

問題: 単一チャンネルの画像が torchvision.utils.make_grid を通過した後、3 チャンネルの画像が生成されます。

torch.Size([4, 1, 256, 256]) → torch.Size([3, 518, 1034])

理由: 関数内に次のコードがあります。

 合成画像が 1 チャンネルの場合、画像は 1 チャンネルに 3 回重畳されます。

解決策: コードを次のように変更します。