matplotlibで描画したイメージをnumpy配列で取得する方法

Question

plt上で描画したイメージをnumpy配列で取得したいのですが
pltからImageオブジェクトを取得する方法がわからず困っています。

import matplotlib.pyplot as plt
import matplotlib.patches as patches
import numpy as np

ax = plt.axes()

# 図形を描画
ax.add_patch(patches.Rectangle(xy=(0, 0), width=400, height=300, fc='w', fill=False))
ax.add_patch(patches.Circle(xy=(50,50), radius=20, fc='g'))
ax.add_patch(patches.Circle(xy=(100,100), radius=20, fc='g'))
ax.add_patch(patches.Circle(xy=(150,150), radius=20, fc='g'))

plt.axis('scaled')

# numpyアレイに変換（したい）
imgarray = np.array(ax) # ←もちろん動かない
print(imgarray)

plt.show()

画像データであればPILで読み込めると思いますが
一度画像で保存して読み込みなどは処理速度的に苦しいです。

from PIL import Image
imgarray = np.array(Image.open('hoge.png'))

matplotlibでやる必要もないのですが
python上で描画した図形をnumpy配列で取得する良い方法はないでしょうか。

何卒ご教示頂きたく、よろしくお願いいたします。

Answer 1

OpenCVでやれば何も苦労しませんでした。

import cv2

img = np.zeros((400,300,3), np.uint8)
img = cv2.circle(img,(100,100), 20, (255,0,0), -1)
img = cv2.circle(img,(200,200), 20, (255,0,0), -1)

cv2.imshow("image", img)
cv2.waitKey(0)