pro sampling avaiable, updated cvae

2022-09-22 16:32:37 +00:00
parent c5ae37aacc
commit a124f95f1f
4 changed files with 297 additions and 118 deletions
@@ -174,7 +174,7 @@ def sample(folder_main, folder_curv):
        uv_curvature = tf.math.argmax(pred, axis=-1)
        uv_curvature = tf.one_hot(uv_curvature, 7)

-        pred = model.sample(batch['uv_static'][0], 
+        pred = model.sample_single(batch['uv_static'][0], 
                            batch['uv_body_posed'][0], 
                            uv_curvature, 
                            batch['uv_mask'][0],