fix body model

configs/interact_teach.yaml

@@ -22,6 +22,9 @@ samples: 1
 logger_level: INFO
 slerp_ws: 8
 
+# joints / vertices / smpl
+repr_type: 'smpl'
+
 # Composing nested config with default
 defaults:
   - data: null
@@ -35,4 +38,4 @@ defaults:
   - _self_
 
 
-data.batch_size: 1
\ No newline at end of file
+data.batch_size: 1

requirements.txt

@@ -44,7 +44,7 @@ moviepy==1.0.3
 multidict==6.0.2
 networkx==2.7.1
 numpy==1.22.3
-oauthlib==3.2.0
+oauthlib>=3.2.1
 omegaconf==2.1.1
 opencv-python==4.5.5.64
 packaging==21.3

teach/model/motiondecoder/actor.py

@@ -40,7 +40,7 @@ class ActorAgnosticDecoder(pl.LightningModule):
         latent_dim = z.shape[1]
         bs, nframes = mask.shape
         nfeats = self.hparams.nfeats
-
+        
         z = z[None]  # sequence of 1 element for the memory
 
         # Construct time queries

teach/render/mesh_viz.py

@@ -58,7 +58,7 @@ def visualize_meshes(vertices, pcd=None, multi_col=None,
             pcd = pcd.unsqueeze(0)
     mesh_rec = vertices
     smpl = get_body_model(path=f'{get_original_cwd()}/data/smpl_models',
-                          model_type='smpl', gender='neutral',
+                          model_type='smpl', gender='male',
                           batch_size=1, device='cpu')
     
     mv = MeshViewer(width=im_width, height=im_height,