run-35

2022-06-27 12:27:55 +00:00
parent 5e00b395ad
commit cd498e79f3
11 changed files with 301 additions and 109 deletions
@@ -105,8 +105,8 @@ class Generator(tf.keras.Model):

        image_out = self.synthesis(w_broadcasted)
        # image_out = image_out + conditions[2]
-        image_out = image_out + self.zero_body[:image_out.shape[0]]
-        image_out = image_out * conditions[3][:,tf.newaxis,:,:]
+        image_out = image_out + conditions['uv_shaped_body']
+        image_out = image_out * conditions['uv_mask'][:,tf.newaxis,:,:]

        if ret_w_broadcasted:
            return image_out, w_broadcasted