github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-11 07:37:04 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

update xlib/image/ImageProcessor.py

Browse source
This commit is contained in:
iperov 2021-10-06 14:51:23 +04:00
parent 65fb01ff39
commit 09400dce75
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
xlib/image/ImageProcessor.py
View file

@ -374,7 +374,7 @@ class ImageProcessor:
self._img = img self._img = img
return self return self
def to_grayscale(self) -> 'ImageProcessor': def to_grayscale(self) -> 'ImageProcessor':
""" """
Converts 3 ch bgr to grayscale. Converts 3 ch bgr to grayscale.
@ -383,18 +383,18 @@ class ImageProcessor:
_,_,_,C = img.shape _,_,_,C = img.shape
if C != 1: if C != 1:
dtype = self.get_dtype() dtype = self.get_dtype()
if C == 2: if C == 2:
img = img[...,:1] img = img[...,:1]
elif C >= 3: elif C >= 3:
img = img[...,:3] img = img[...,:3]
img = np.dot(img, np.array([0.1140, 0.5870, 0.2989], np.float32)) [...,None] img = np.dot(img, np.array([0.1140, 0.5870, 0.2989], np.float32)) [...,None]
img = img.astype(dtype) img = img.astype(dtype)
self._img = img self._img = img
return self return self
def resize(self, size : Tuple, interpolation : 'ImageProcessor.Interpolation' = None, new_ip=False ) -> 'ImageProcessor': def resize(self, size : Tuple, interpolation : 'ImageProcessor.Interpolation' = None, new_ip=False ) -> 'ImageProcessor':
""" """
resize to (W,H) resize to (W,H)
@ -492,6 +492,9 @@ class ImageProcessor:
class Interpolation(IntEnum): class Interpolation(IntEnum):
LINEAR = 0 LINEAR = 0
CUBIC = 1 CUBIC = 1
LANCZOS4 = 4
_cv_inter = { ImageProcessor.Interpolation.LINEAR : cv2.INTER_LINEAR, _cv_inter = { ImageProcessor.Interpolation.LINEAR : cv2.INTER_LINEAR,
ImageProcessor.Interpolation.CUBIC : cv2.INTER_CUBIC } ImageProcessor.Interpolation.CUBIC : cv2.INTER_CUBIC,
ImageProcessor.Interpolation.LANCZOS4 : cv2.INTER_LANCZOS4,
}