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

2nd pass of extractor : speed significantly increased

Browse source
This commit is contained in:
iperov 2019-03-08 16:05:24 +04:00
parent 8f7b268376
commit 0cc251a9ab
3 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
facelib/LandmarksExtractor.py
View file

@ -121,7 +121,7 @@ class LandmarksExtractor(object):
predicted = self.keras_model.predict (image)
pts_img = get_pts_from_predict ( predicted[-1][0], center, scale)
pts_img = get_pts_from_predict ( predicted[-1], center, scale)
pts_img = [ ( int(pt[0]), int(pt[1]) ) for pt in pts_img ]
landmarks.append ( ( (left, top, right, bottom),pts_img ) )