github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-16 10:03:42 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix to_grayscale()

Browse source
This commit is contained in:
iperov 2021-07-24 09:09:53 +04:00
parent 2125193b80
commit 5ae2b7641f
1 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

20
xlib/image/ImageProcessor.py
View file

@ -442,6 +442,26 @@ class ImageProcessor:
self._img = img
return self
def to_grayscale(self) -> 'ImageProcessor':
"""
Converts 3 ch bgr to grayscale.
"""
img, xp = self._img, self._xp
_,_,_,C = img.shape
if C != 1:
dtype = self.get_dtype()
if C == 2:
img = img[...,:1]
elif C >= 3:
img = img[...,:3]
img = xp.dot(img, xp.array([0.1140, 0.5870, 0.2989], xp.float32)) [...,None]
img = img.astype(dtype)
self._img = img
return self
def resize(self, size : Tuple, interpolation : 'ImageProcessor.Interpolation' = None, new_ip=False ) -> 'ImageProcessor':
"""
resize to (W,H)