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

-

Browse source
This commit is contained in:
iperov 2019-04-15 09:47:59 +04:00
commit d4a72b49c6
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
facelib/FANSegmentator.py
View file

@ -69,6 +69,7 @@ class FANSegmentator(object):
input_image = input_image[np.newaxis,...] input_image = input_image[np.newaxis,...]
result = np.clip ( self.model.predict( [input_image] ), 0, 1.0 ) result = np.clip ( self.model.predict( [input_image] ), 0, 1.0 )
result[result < 0.1] = 0 #get rid of noise
if input_shape_len == 3: if input_shape_len == 3:
result = result[0] result = result[0]