david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
facelib/MTCExtractor.py
+6 -2
View File
@@ -86,8 +86,12 @@ class MTCExtractor(object):
def __exit__(self, exc_type=None, exc_value=None, traceback=None): def __exit__(self, exc_type=None, exc_value=None, traceback=None):
return False #pass exception between __enter__ and __exit__ to outter level return False #pass exception between __enter__ and __exit__ to outter level
def extract_from_bgr (self, input_image): def extract (self, input_image, is_bgr=True):
input_image = input_image[:,:,::-1].copy()
if is_bgr:
input_image = input_image[:,:,::-1].copy()
is_bgr = False
(h, w, ch) = input_image.shape (h, w, ch) = input_image.shape
input_scale = self.scale_to / max(w,h) input_scale = self.scale_to / max(w,h)