PILで取り込んだ画像をnumpyに変換してmaskと重ねる

普段，私は画像はPIL(Pillow)で取り込んでいます。 色々あって，semantic segmentationを取り組んでいるのですが，やっぱりmaskデータと重ねて描写してみたくなります。 いまいちやり方がわからなかったので，自分なりにググってやってみたことを書いておきます。 単なる備忘録です。 絶対もっといいやり方があると思います。

参考にしたのは，

stackoverflow.com

です。

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

#データ取り込み
image = Image.open(img_data)
mask = np.array(mask_data)

# numpyへの変換
image = np.asarray(image)

# figureサイズ操作
plt.figure(figsize=(XX, XX), dpi=XX)

# 描写
plt.imshow(image)
plt.imshow(mask, "jet", alpha=0.2)

です。 最後の引数"jet"は色の好みで，alphaで透明度を変更できます。