github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-30 11:40:36 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

extractor: fixes, optimizations,

Browse source 
manual extractor: added 'a' option to switch accuracy mode
This commit is contained in:
iperov 2019-03-21 18:56:32 +04:00
parent 63c9be3d1f
commit 7a0cc56603
3 changed files with 121 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

3
facelib/LandmarksExtractor.py
View file

@ -46,7 +46,6 @@ class LandmarksExtractor(object):
predicted = self.keras_model.predict (image).transpose (0,3,1,2)
pts_img = self.get_pts_from_predict ( predicted[-1], center, scale)
pts_img = [ ( int(pt[0]), int(pt[1]) ) for pt in pts_img ]
landmarks.append (pts_img)
except:
landmarks.append (None)
@ -118,4 +117,4 @@ class LandmarksExtractor(object):
c[i] += np.sign(diff)*0.25
c += 0.5
return [ self.transform (c[i], center, scale, a.shape[2]) for i in range(a.shape[0]) ]
return np.array( [ self.transform (c[i], center, scale, a.shape[2]) for i in range(a.shape[0]) ] )