github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactorings, optimizations

Browse source
This commit is contained in:
iperov 2019-03-21 11:58:54 +04:00
parent 97008291b8
commit d1676dea06
4 changed files with 171 additions and 130 deletions
Download patch file Download diff file Expand all files Collapse all files

8
facelib/S3FDExtractor.py
View file

@ -19,8 +19,12 @@ class S3FDExtractor(object):
def __exit__(self, exc_type=None, exc_value=None, traceback=None):
return False #pass exception between __enter__ and __exit__ to outter level
def extract_from_bgr (self, input_image):
input_image = input_image[:,:,::-1].copy()
def extract (self, input_image, is_bgr=True):
if is_bgr:
input_image = input_image[:,:,::-1]
is_bgr = False
(h, w, ch) = input_image.shape
d = max(w, h)