ag/pixelocr
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Simplify Image.color() and make in return ints instead of floats.

Browse source
This commit is contained in:
Andrey Golovizin 2014-09-02 14:33:39 +02:00
parent af351617d8
commit dac3475579
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
pixelocr/image.py
View file

@ -18,7 +18,6 @@ import itertools
from io import BytesIO from io import BytesIO
import numpy as np import numpy as np
from scipy.stats.mstats import mode
from skimage.io import imread, imsave from skimage.io import imread, imsave
from .utils import cached_property, pairwise from .utils import cached_property, pairwise
@ -170,8 +169,8 @@ class Image(object):
return None return None
mask3 = np.dstack([~self.bitmap] * 3) mask3 = np.dstack([~self.bitmap] * 3)
colors = np.ma.MaskedArray(self.data, mask3).reshape(-1, 3) colors = np.ma.MaskedArray(self.data, mask3).reshape(-1, 3)
modes, counts = mode(colors) r, g, b = colors.mean(axis=0)
return tuple(modes[0]) return int(r), int(g), int(b)
def serialize(self): def serialize(self):
"""Serialize the image as some hashable object.""" """Serialize the image as some hashable object."""