IPython.core.display.Image を numpy.ndarrayにしたい．

Question

前提

Jupyter lab Desktop (Version 3.5.0) 環境下で，
matplotlibを使って画像付きのグラフを作成しています．

実現したいこと

Ipython.core.display.Image オブジェクトをnumply.ndarrayに変換したい．

↓は実際に扱うオブジェクトではありませんがこれを

from IPython.display import Image
img = Image(data='https://upload.wikimedia.org/wikipedia/commons/thumb/e/e3/Cherry_blossoms_%282004%29.jpg/300px-Cherry_blossoms_%282004%29.jpg',
     retina=True)
type(img)
# IPython.core.display.Image

↓これのような形式に変換したいのです．

import matplotlib.pyplot as plt
img_plt = plt.imread('hoge.png')
type(img_plt)
# numpy.ndarray

真の目的としてはOffsetImageメソッドの引数に画像データを入れたいのです．
numpy.ndarray，もしくはPIL.PngImagePlugin.PngImageFileでないとエラーがでてしまいます．

from matplotlib.offsetbox import OffsetImage
OffsetImage(img)

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

IPython.core.display.Imageを引数にいれた場合

TypeError: Image data of dtype object cannot be converted to float

Answer 1

class IPython.core.display.Image(data=None, url=None, filename=None, format=u'png', embed=False)

data : unicode, str or bytes
The raw data or a URL to download the data from.

とのことなので以下のような感じでdataからPIL画像を作成できました。

from IPython.display import Image
from PIL import Image as PILImage
from io import BytesIO

img = Image(data='https://upload.wikimedia.org/wikipedia/commons/thumb/e/e3/Cherry_blossoms_%282004%29.jpg/300px-Cherry_blossoms_%282004%29.jpg',
     retina=True)

pil_img = PILImage.open(BytesIO(img.data))

from matplotlib.offsetbox import OffsetImage
off_img = OffsetImage(pil_img)
print(type(off_img)) # <class 'matplotlib.offsetbox.OffsetImage'>

Comments

[YaKa]

うわぁ！ありがとうございます．上記を導入しましたら想定の動作をしてくれました！コードの中身は勉強します．