Update src/app/process_data.py

src/app/process_data.py

@@ -31,6 +31,7 @@ def get_a_sample(mo_data, motion_len=6, SEQLEN=16, smplx_pth=None):
     global_orient = mo_data['global_orient']   # L,3
     body_pose = mo_data['body_pose']           # L,63 -> L,21,3
     length = transl.shape[0]
+    print("Get a sample")
      
     if (length - (SEQLENTIMES2-2)*motion_len) <= 0:
         return None
@@ -72,5 +73,5 @@ def get_a_sample(mo_data, motion_len=6, SEQLEN=16, smplx_pth=None):
                             global_orient=torch.tensor(global_orient_all, dtype=torch.float32),
                             transl=torch.tensor(transl_all, dtype=torch.float32),
                             ).joints[:, SELECTED_JOINT28]
-        
+    print("Get a sample returns successfully!")
     return joints.reshape(motion_len, SEQLEN, 28, 3) # a Tensor of size (6, 16, 28, 3)