david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
facelib/FANExtractor.py
+7 -1
View File
@@ -63,7 +63,13 @@ class FANExtractor(object):
images += [ self.crop(input_image, c, scale) ] images += [ self.crop(input_image, c, scale) ]
images = np.stack (images) 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): for i, pred in enumerate(predicted):
ptss += [ self.get_pts_from_predict ( pred, centers[i], scale) ] ptss += [ self.get_pts_from_predict ( pred, centers[i], scale) ]