train encoder on raw curvature values

2023-01-27 11:13:10 +00:00
parent a3c5b526b8
commit 6e6b0a6604
8 changed files with 60 additions and 71 deletions
@@ -190,13 +190,7 @@ class DecoderNew(tf.keras.Model):

        x = self.decoder(x, training=training)

-        if self.is_label:
-            x = x * mask[:,:,:,tf.newaxis]
-
-            background = tf.ones((mask.shape[0],512,256,1)) * (1 - mask[:,:,:,tf.newaxis])
-            x = tf.concat((background, x), axis=-1)
-
-            x = tf.math.softmax(x)
+        # if self.is_label:

        if len(inputs) == 2:
            if len(mask.shape) == 2: