github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-24 15:16:21 -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

@ -21,6 +21,9 @@ class DLIBExtractor(object):
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()
(h, w, ch) = input_image.shape