tensor化した画像を表示させたい

Question

前提・実現したいこと

numpyをtorch.tensorを用いてtensorに変換しました。
変換したtensorから画像を表示させたいです。

発生している問題・エラーメッセージ

試したこと

for i in X_train:
i=i.numpy()
plt.imshow(i)
tensorをnumpyに戻し表示させようと考えたのですがInvalid shape (3, 96, 96) for image dataのエラーが出て表示ができませんでした。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[jbpb0]

https://qiita.com/takurooo/items/e356dfdeec768d8f7146
の「Channel First」の最後のところを見てください

[uram]

ありがとうございます。transpose(1,2,0)で読み込みには成功したのですが、オリジナルと同様にuint8に変換後imshowで表示を行うと薄っすらと輪郭はわかるのですが、虹色に表示されます。この解決方法がわかりません。

[uram]

自己解決しました。numpyに変換後reshape(96,96,3)を行い表示ができました。

Answer 1

for i in X_train:
i=i.numpy()
i=i.reshape(96,96,3)
plt.imshow(i)