activate all levels

2023-03-15 22:20:33 +01:00
parent e108460c48
commit 1373b363b5
2 changed files with 99 additions and 89 deletions
@@ -416,8 +416,8 @@ class PyramidNew(tf.keras.Model):
            pred = pred + levels_input[0][2] # uv_body_posed_32
        if 1 != len(self.levels):
            pred = self.levels[0].enhance(pred, levels_input[0][-2], training=training)  # levels_input[i][-2]=uv_mask_res
-            pred = tf.image.resize(pred, (self.levels[1].size, self.levels[1].size//2), tf.image.ResizeMethod.BILINEAR)
            out_levels.append([pred, mean, logvar])
+            pred = tf.image.resize(pred, (self.levels[1].size, self.levels[1].size//2), tf.image.ResizeMethod.BILINEAR)
            pred = pred * levels_input[1][-2][:,:,:,tf.newaxis]
        else:
            # pred = self.enhance(pred, levels_input[0][-2], training=training)