前提
モノクロ画像のチャンネル数1をRGBのチャンネル数3に変更できず困っております。
実現したいこと
事前学習済みのResNetを使用したいのですが入力のカラーチャンネルはRGBの3にする必要がありました。しかし私が使用する画像はモノクロ画像でカラーチャンネルが1です。このカラーチャンネルを1から3へ変更したいです。
発生している問題・エラーメッセージ
Given groups=1, weight of size [64, 3, 7, 7], expected input[2, 1, 32, 32] to have 3 channels, but got 1 channels instead
該当のソースコード
PyTorch
1net = net.to(device) 2summary(net,(1,32,32))
試したこと
試したこととしてdata_transportを行いチャンネル数を変更しようとしたのですが、うまくいきませんでした。
補足情報(FW/ツールのバージョンなど)
python3.8.10
参考
https://teratail.com/questions/204911
ありがとうございます。
いただいたコードをもとに行ったところ以下のようになったのですが、Numpyを使用したものは変更できないのでしょうか。もしご存知でしたら教えていただきたいです。
'numpy.ndarray' object has no attribute 'convert'
> Numpyを使用したものは変更できないのでしょうか。
opencvはnumpy配列を扱えるので、opencvの「cv2.cvtColor」で「cv2.COLOR_GRAY2RGB」を指定したら変換できると思います
参考
https://qiita.com/7vvXi/items/331fa6da1a75b4134cf5