Errorが出ます。Google Colab とPyTorchで Google Driveのオリジナル画像を読み込んで、表示するコード

Question

AIを勉強したい初心者ですが、最初から躓き。
windows 10
Google Colaboratory
Python
MyDriveの下に、オリジナルのカラー画像を３枚置きました。
これを読み込んで、表示したいのですが、Errorがでました。
見よう見まねのコードを入れたのですが、数値は羅列されても、画像はでません。
どなたか基本も含めて、ご指導ください。

TypeError: Invalid shape (3,616,820) for image data

オリジナルデータは横820*縦616の画素
カラー画像：　tri1.jpg tri2.jpg tri3.jpg

import os
import torch
import torchvision
import torchvision.transforms as transforms
from PIL import Image
import matplotlib.pyplot as plt

fnames=os.listdir('/content/drive/MyDrive/test')
for fname in fnames:
  image=Image.open('/content/drive/MyDrive/test/tri1.jpg')

  loader=transforms.Compose([
  
  transforms.ToTensor()
])

image=loader(image)
print(image)
plt.imshow(image)
plt.show()

Comments

[jbpb0]

pythonのコードの一番最初の行のすぐ上に
```python
だけの行を追加してください
また、pythonのコードの一番最後の行のすぐ下に
```
だけの行を追加してください

現状、コードがとても読み辛いです
質問にコードを載せる際に上記をやってくれたら、他人がコードを読みやすくなり、コードの実行による現象確認もやりやすくなるので、回答されやすくなります

[meg_]

plt.imshow(image[0])とかで表示できますが、そういうことですか？

[hiruandon3]

jbpb0様
　回答が得やすい、質問投稿の様式についてのアドバイスありがとうございます。
　回答をいただくことが目的ですから、質問の意図が伝わる表記方法は大切です。　　

　　・・・・pytｈoh
import os
import torch
:
:
plt.show()
.....

以後、ご指摘のこのような表記とします。

　　ありがとうございます。
　　今後もご指導お願いします。

[hiruandon3]

meg_様

　確認のご質問ありがとうございます。対応が遅くなって申し訳ありません、
　Google Driveの中に保存してあるカラーのオリジナル画像（jpg）を３つ順次読み込んで画像そのものを表示表示し、確認したいと思っています。オリジナル画像は現在
　/content/drive/MyDrive/test/
　のフォルダーにあります。
　現在はとりあえず画像は3枚ですが、最終的にはかなりの枚数の画像を処理したいと考えています。

　投稿した現時点のコードには、不要なライブラリー、モジュールなど入っているかもしれません。　　
　また、不適当な関数、パレメータ指定があるでしょう。希望していない画像（とおもわれる）の数値データは表示されましたが、肝心な画像が表示されません。画像が表示されないと正しく機能しているのか不安です。
　不要なコード、必要なコード、パラメータなどご指導いただけたら嬉しいです。

　meg_様のご指摘の
　plt.imshow(image[0])
　をそのまま一行挿入すればいいのでしょうか。ほかに削除、追加は必要でしょうか。
　基本的な質問ですみません。
　よろしくお願いします。

[jbpb0]

> 以後、ご指摘のこのような表記とします。

まだ直ってません

下記を見て、それを参考にして質問を編集してください
https://teratail-v2.storage.googleapis.com/uploads/contributed_images/e378967f71829feb9e6187e5be60349f.gif
ただし、最後「c++」と記入してるところは「python」で

[jbpb0]

何でそのままやってくれないのかなぁ

.....Python
ではなくて
```python
です

...........
ではなくて
```
です

文字を変えたらダメです
文字数を変えてもダメです

自分でキーボードで入力するのではなく、私のコメントからコピペしてください

[hiruandon3]

修正しました。コピペしました。すみません。
ご指導ありがとうございます。

[jbpb0]

pythonのコードの一番最後の行のすぐ下に
```
「だけ」の行を追加してください

現状、下記のようにコードの行末に付いてます
image=loader(image) print(image) plt.imshow(image) plt.show() ```

[hiruandon3]

jbpb0様
　再度のご指摘ありがとうございます。
　改行しました。

[jbpb0]

コードが見やすくなりました

[hiruandon3]

jbpb0様
　ご指導ありがとうございました。
　今頃ですが気がつきました。
　アドバイスいただいた表記にすれば、質問の欄において、コードだけが別表示に自動的になるようですね。細部に気が回らずご迷惑おかけしました。これに懲りずにご指導いただけば嬉しいです。

Answer 1

画像を表示させるだけなら、PyTorchの機能は要らないので、これだけでできるはず

import os
from PIL import Image
import matplotlib.pyplot as plt

file_path = '/content/drive/MyDrive/test'
fnames = os.listdir(file_path)
for fname in fnames:
  print(fname)
  image = Image.open(file_path + '/' + fname)
  plt.imshow(image)
  plt.show()

　
PyTorchの「transforms.ToTensor()」でテンソルに変換したものを表示させる場合は、変換するときに順番が変わってるので、「np.transpose()」で順番を元に戻す必要があります
下記のコードでは、変換の前後のそれぞれで「shape」を表示させてるので、それを比べてみてください

import numpy as np
import os
import torchvision.transforms as transforms
from PIL import Image
import matplotlib.pyplot as plt

file_path = '/content/drive/MyDrive/test'
fnames = os.listdir(file_path)
for fname in fnames:
  print(fname)
  image = Image.open(file_path + '/' + fname)
  print(np.array(image).shape)
  loader = transforms.Compose([transforms.ToTensor()])
  image = loader(image)
  print(image.shape)
  plt.imshow(np.transpose(image.numpy(), (1,2,0)))
  plt.show()

参考
pyTorchのtransforms,Datasets,Dataloaderの説明と自作Datasetの作成と使用
の「5. Datasetの使い方とDatasetの自作」の「5-1-1. transformsによる前処理の定義」

Comments

[hiruandon3]

jbpb0様
　具体的なコードも提案いただき、エラーも解消し、こちらの希望する処理が出来ました。しかも、今後必要となる処理にまで言及して指導いただきました。
　ありがとうございます。