commit all

unpose_one_hot.py

@@ -14,7 +14,7 @@ from tqdm import tqdm
 from custom_utils.custom_utils import maskFacesBatch, _maskFaces
 
 
-os.environ["CUDA_VISIBLE_DEVICES"] = "1"
+os.environ["CUDA_VISIBLE_DEVICES"] = "2"
 import tensorflow as tf
 gpus = tf.config.experimental.list_physical_devices('GPU')
 if gpus:
@@ -136,7 +136,7 @@ def smpl_run_backward_wrapper(Rt, J, posed):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/data/dataset/test_skirt')
+    parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/data/dataset/train_skirt')
     parser.add_argument("-s", "--start", help = "start of range", default=0)
     parser.add_argument("-e", "--end", help = "end of range", default=1500)
     parser.add_argument("-n", "--name", help = "end of range", default='lhunor')
@@ -174,9 +174,10 @@ if __name__ == '__main__':
 
         if is_skirt:
             pose_dress = np.copy(pose)
-            p_avg = (pose[:, 3:6] + pose[:, 6:9]) / 2
+            pose_dress = np.transpose(pose_dress, [1,0])
+            p_avg = (pose_dress[:, 3:6] + pose_dress[:, 6:9]) / 2
             pose_dress[:,3:9] = np.concatenate([p_avg, p_avg], axis=-1)
-            R = smpl.run_rodrigues(np.transpose(pose_dress, [1,0])).numpy()
+            R = smpl.run_rodrigues(pose_dress).numpy()
             for b in range(frames):
                 R[b][[4,5,7,8,10,11],:,:] = np.eye(3)
         else:
@@ -184,6 +185,8 @@ if __name__ == '__main__':
 
         Rt = tf.transpose(R, [0, 1, 3, 2])
         T_body, J, _ = smpl_run_wrapper(R, shape)
+
+        #test
         # zero_body = smpl_run_wrapper(smpl.run_rodrigues(np.zeros((1, 72), np.float32)), shape[:1, :])[0][0].numpy()