save

2023-03-06 09:55:48 +00:00
parent dfdf8d408e
commit 0b518e7f0e
46 changed files with 3392 additions and 160 deletions
@@ -79,16 +79,17 @@ class CVAE(tf.keras.Model):

        return z_cond

+
    def sample(self, uv_template, uv_body_posed, uv_mask, noise=None):
        if noise is None:
-            eps = tf.random.normal(shape=(uv_template.shape[0], self.latent_dim))
+            eps = tf.random.normal(shape=(uv_template.shape[0], 8, 4, self.latent_dim))
        else:
            eps = noise

        z_cond = self.encode_conds(uv_template, uv_body_posed, training=False)
        z = tf.concat([eps, z_cond], axis=-1)

-        x = self.decoder(z, training=False)
+        x = self.decoder([z, uv_mask], training=False)

        # x = x * uv_mask[:,:,:,tf.newaxis]