run-35

custom_utils/custom_utils.py

@@ -135,4 +135,16 @@ def plot_3d(mesh_gt, mesh_pred_unposed, faces, collision, collision_faces, colli
                         zaxis = dict(nticks=4, range=[-1,1],),)
                         )
 
-    return fig
+    return fig
+
+
+def batch_to_dict(batch, signature):
+    result = dict()
+    for i, key in enumerate(signature.keys()):
+            result[key] = batch[i]
+    return result
+
+conditions = ['pose_shape', 'uv_static', 'uv_mask', 'uv_shaped_body']
+def nest_conditions(batch):
+    batch['conditions'] = {d: batch.pop(d) for d in conditions}
+    return batch