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

extractor: fix for amd

Browse source
This commit is contained in:
Colombo 2019-10-27 12:47:12 +04:00
parent f17a54b23a
commit 8035325f92
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
facelib/FANExtractor.py
View file

@ -63,7 +63,13 @@ class FANExtractor(object):
images += [ self.crop(input_image, c, scale) ]
images = np.stack (images)
predicted = self.model.predict (images.astype(np.float32) / 255.0).transpose (0,3,1,2)
images = images.astype(np.float32) / 255.0
predicted = []
for i in range( len(images) ):
predicted += [ self.model.predict ( images[i][None,...] ).transpose (0,3,1,2)[0] ]
predicted = np.stack(predicted)
for i, pred in enumerate(predicted):
ptss += [ self.get_pts_from_predict ( pred, centers[i], scale) ]