added skinning

custom_utils/custom_utils.py

@@ -89,8 +89,8 @@ def maskFacesBatch(V, F, mask):
     return Vs, Fs
 
 
-def plot_3d(mesh_gt, mesh_pred, faces, collision, collision_faces, collision_color):
-    mesh_pred[:,1] += 1
+def plot_3d(mesh_gt, mesh_pred_unposed, faces, collision, collision_faces, collision_color):
+    mesh_pred_unposed[:,1] += 1
     collision[:,1] += 2
     fig = go.Figure(data=[
         go.Mesh3d(
@@ -105,9 +105,9 @@ def plot_3d(mesh_gt, mesh_pred, faces, collision, collision_faces, collision_col
         ),
         go.Mesh3d(
             # 8 vertices of a cube
-            x=mesh_pred[:,0],
-            y=mesh_pred[:,1],
-            z=mesh_pred[:,2],
+            x=mesh_pred_unposed[:,0],
+            y=mesh_pred_unposed[:,1],
+            z=mesh_pred_unposed[:,2],
             # i, j and k give the vertices of triangles
             i = faces[:,0],
             j = faces[:,1],