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

_

Browse source
This commit is contained in:
iperov 2019-04-15 10:03:44 +04:00
commit 2db46007cd
2 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
facelib/FANSegmentator.py
View file

@ -49,7 +49,8 @@ class FANSegmentator(object):
io.log_err("Unable to load VGG11 pretrained weights from vgg11_enc_weights.npy") io.log_err("Unable to load VGG11 pretrained weights from vgg11_enc_weights.npy")
if training: if training:
self.model.compile(loss='mse', optimizer=Adam(tf_cpu_mode=2)) #self.model.compile(loss='mse', optimizer=Adam(tf_cpu_mode=2))
self.model.compile(loss='binary_crossentropy', optimizer=Adam(tf_cpu_mode=2), metrics=['accuracy'])
def __enter__(self): def __enter__(self):
return self return self

4
models/Model_FANSEG/Model.py
View file

@ -56,9 +56,9 @@ class Model(ModelBase):
def onTrainOneIter(self, generators_samples, generators_list): def onTrainOneIter(self, generators_samples, generators_list):
target_src, target_src_mask = generators_samples[0] target_src, target_src_mask = generators_samples[0]
loss = self.fan_seg.train_on_batch( [target_src], [target_src_mask] ) loss,acc = self.fan_seg.train_on_batch( [target_src], [target_src_mask] )
return ( ('loss', loss), ) return ( ('loss', loss), ('acc',acc))
#override #override
def onGetPreview(self, sample): def onGetPreview(self, sample):