plt.showで表示させた画像を保存したい。

Question

前提・実現したいこと

plt.showで表示させた画像を保存したいが以下のような問題が起きてしまい困っています。
解決策を教えていただけますと幸いです。

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

X_testの画像を表示させ、それを保存しようとすると、plt.imshowで表示された画像とは異なる以下のような画像が書き出されてしまう状態です。元々表示させた画像は、部屋の内装の写真になります。

該当のソースコード

from PIL import Image
import os, glob
import numpy as np
from sklearn import model_selection
import matplotlib.pyplot as plt

classes = ["concrete","tile","mortar","brick","wood"]
num_classes = len(classes)
image_size = 50

for index, classlabel in enumerate(classes):
,,,,photos_dir = "./" + classlabel
,,,,files = glob.glob(photos_dir + "/*.jpg")
,,,,for i, file in enumerate(files):
,,,,,,,,if i >= 500: break
,,,,,,,,image = Image.open(file)
,,,,,,,,image = image.convert("RGB")
,,,,,,,,image = image.resize((image_size, image_size))
,,,,,,,,X.append(image)
,,,,,,,,Y.append(index)

X = X
Y = Y

X_train, X_test, y_train, y_test = model_selection.train_test_split(X, Y)
xy = (X_train, X_test, y_train, y_test)

plt.imshow(X_test[0])
plt.show()
plt.save("a.jpg",X_test[0])

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

python

Answer 1

plt.show()で表示される画像と同じ画像を保存したい場合、以下の二点が間違っています。

問題点１　順序

plt.show() の後で保存することはできません。
保存した後、plt.show()で表示することはできます。

問題点２ 保存のメソッドと引数

plt.save("a.jpg",X_test[0])

そういうメソッドはありません。引数の数も間違っています。

plt.savefig("a.jpg")

です。

Comments

[Shinji08]

コメントありがとうございます。
問題点２についてなんですが、私の記述が間違っておりplt.imsave("a.jpg",X_test[0])でした。
この場合でも間違いでしょうか？

[Shinji08]

無事解決しました。ありがとうございます。