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

Add DLIBExtractor.extract

Browse source 
Fix AttributeError: 'DLIBExtractor' object has no attribute 'extract'
This commit is contained in:
tyousou 2019-04-16 20:34:35 +08:00 • committed by GitHub
commit c48e472a00
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
facelib/DLIBExtractor.py
View file

@ -20,6 +20,9 @@ class DLIBExtractor(object):
def __exit__(self, exc_type=None, exc_value=None, traceback=None):
del self.dlib_cnn_face_detector
return False #pass exception between __enter__ and __exit__ to outter level
def extract(self, input_image, is_bgr=True):
return self.extract_from_bgr(input_image)
def extract_from_bgr (self, input_image):
input_image = input_image[:,:,::-1].copy()