create end2end laplacian pyramid

2023-01-26 20:39:54 +00:00
parent 8c0436cea8
commit 0dfe7f7659
5 changed files with 133 additions and 116 deletions
@@ -45,7 +45,7 @@ class UV2Mesh(tf.keras.Model):
        if uv.shape[1] == 3:
            uv = tf.transpose(uv, [0,2,3,1])
        if uv.shape[1] != 512:
-            uv = tf.image.resize(uv, (512, 256), tf.image.ResizeMethod.LANCZOS5)
+            uv = tf.image.resize(uv, (512, 256), tf.image.ResizeMethod.BILINEAR)
            
        batch_size = uv.shape[0]