From af0b3904fcd3269832b7f622785fd1dc45291e29 Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 15 Apr 2021 21:43:47 +0400 Subject: [PATCH] XSeg trainer: additional dssim loss. --- models/Model_XSeg/Model.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/models/Model_XSeg/Model.py b/models/Model_XSeg/Model.py index 4ab23fc..a257c11 100644 --- a/models/Model_XSeg/Model.py +++ b/models/Model_XSeg/Model.py @@ -95,6 +95,9 @@ class XSegModel(ModelBase): gpu_pred_list.append(gpu_pred_t) gpu_loss = tf.reduce_mean( tf.nn.sigmoid_cross_entropy_with_logits(labels=gpu_target_t, logits=gpu_pred_logits_t), axis=[1,2,3]) + gpu_loss += tf.reduce_mean ( 0.1*nn.dssim(gpu_target_t, gpu_pred_logits_t, max_val=1.0, filter_size=3), axis=[1]) + + gpu_losses += [gpu_loss] gpu_loss_gvs += [ nn.gradients ( gpu_loss, self.model.get_weights() ) ]