commit all

2023-01-25 12:25:59 +00:00
parent 1bf224841f
commit 7470ca9fb6
139 changed files with 82069 additions and 811 deletions
@@ -60,13 +60,13 @@ class Pyramid(tf.keras.Model):
        else:
            self.avg_garment = None

-    def enhance(self, low, mask, training=None):
+    def enhance(self, low, mask, training):
        x = self.upscale_enhancer(low, training=training)
        x = x * (1 - mask[:,:,:,tf.newaxis])
        low = x + low
        return low

-    def call(self, inputs, training=None):
+    def call(self, inputs, training):
        uv_unposed, uv_static, uv_body_posed, uv_curv, uv_mask, uv_unposed_low, uv_static_low, uv_body_posed_low, uv_curv_low, uv_mask_low = inputs

        if self.use_full_res_encoders: