Python matplotlibで画像を保存

Question

前提・実現したいこと

matplotlibで画像を保存を保存したいです．ただし，何度も呼び出すため，画像を表示させることなく保存したいです．
ご教授のほどよろしくお願いいたします．

該当のソースコード

import matplotlib.pyplot as plt
import cv2
import matplotlib.patches as mpatches

def draw(filename1, filename2, filename3, x, y):
    img1 = cv2.imread(filename1)
    img2 = cv2.imread(filename2)
    fig, ax = plt.subplots(2,1, figsize=(10,10))
    circle1 = mpatches.Circle(xy=(x, y), radius=5)
    ax[0].add_patch(circle1)
    circle2 = mpatches.Circle(xy=(x, y), radius=5)
    ax[1].add_patch(circle2)
    
    # 以下のimshowを使わずに画像を読み込んで保存したいです
    ax[0].imshow(img1)
    ax[1].imshow(img2)

    plt.savefig(filename3)

Comments

[tiitoi]

質問のコードでウィンドウに画像が表示されるのですか？

[tthk]

jupyter notebookで実行しているので，セルの下に表示されます．

Answer 1

plt.savefig() の次の行に plt.close() を追記してみたらどうでしょうか。

import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline

x = np.arange(2*np.pi)
ax = plt.subplot()
ax.plot(x,np.sin(x))
plt.savefig('out.png')
plt.close()
# ↑これで表示されずにセーブできるはず

Comments

[magichan]

完全にかぶりました

[tthk]

できました！
ありがとうございます．

Answer 2

jupyter notebookで実行しているので，セルの下に表示されます．

plt.savefig(filename3) の下に plt.close() を入れると、表示されなくなると思います。

Comments

[tthk]

できました！
ありがとうございます．