diff --git a/pixelocr/image.py b/pixelocr/image.py index d115486..6f2e407 100644 --- a/pixelocr/image.py +++ b/pixelocr/image.py @@ -7,13 +7,20 @@ from skimage.color import rgb2gray from .utils import cached_property -class Image(object): +class Image(IPythonImageMixin): + """Basic image class.""" + def __init__(self, data): self._data = data def __getitem__(self, key): return type(self)(self._data.__getitem__(key)) + def _repr_png_(self): + buf = BytesIO() + imsave(buf, self._data) + return buf.getvalue() + @classmethod def fromfile(cls, filename): return cls(imread(filename)) @@ -30,11 +37,6 @@ class Image(object): def T(self): return type(self)(self._data.swapaxes(0, 1)) - def _repr_png_(self): - buf = BytesIO() - imsave(buf, self._data) - return buf.getvalue() - @property def height(self): return self._data.shape[0]