train on template only with mask applied

2022-06-14 10:16:55 +00:00
parent def8a85cf9
commit 245f811d47
11 changed files with 112 additions and 60 deletions
@@ -102,6 +102,7 @@ class Generator(tf.keras.Model):
            w_broadcasted = self.truncation_trick(w_broadcasted, truncation_psi, truncation_cutoff)

        image_out = self.synthesis(w_broadcasted)
+        image_out = image_out * conditions[3][:,tf.newaxis,:,:]

        if ret_w_broadcasted:
            return image_out, w_broadcasted