diff --git a/facelib/FANSegmentator.py b/facelib/FANSegmentator.py index fb7f4d5..e2ae705 100644 --- a/facelib/FANSegmentator.py +++ b/facelib/FANSegmentator.py @@ -50,7 +50,7 @@ class FANSegmentator(object): if training: #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']) + self.model.compile(loss='binary_crossentropy', optimizer=Adam(tf_cpu_mode=2) ) def __enter__(self): return self diff --git a/models/Model_DEV_FANSEG/Model.py b/models/Model_DEV_FANSEG/Model.py index 2233947..f84345c 100644 --- a/models/Model_DEV_FANSEG/Model.py +++ b/models/Model_DEV_FANSEG/Model.py @@ -65,9 +65,9 @@ class Model(ModelBase): def onTrainOneIter(self, generators_samples, generators_list): target_src, target_src_mask = generators_samples[0] - loss,acc = self.fan_seg.train_on_batch( [target_src], [target_src_mask] ) + loss = self.fan_seg.train_on_batch( [target_src], [target_src_mask] ) - return ( ('loss', loss), ('acc',acc)) + return ( ('loss', loss), ) #override def onGetPreview(self, sample):