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

fix

Browse source
This commit is contained in:
iperov 2019-03-21 12:15:04 +04:00
parent d1676dea06
commit 63c9be3d1f
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
facelib/MTCExtractor.py
View file

@ -86,8 +86,12 @@ class MTCExtractor(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].copy()
is_bgr = False
(h, w, ch) = input_image.shape
input_scale = self.scale_to / max(w,h)