Updated motion processing scripts

Author: Svito-zar

data_processing/bvh2features.py

@@ -63,9 +63,9 @@ def extract_joint_angles(bvh_dir, files, dest_dir, pipeline_dir, fps):
 
     # Setup parameter parser
     parser = ArgumentParser(add_help=False)
-    parser.add_argument('--bvh_dir', '-orig', default="./data/",
+    parser.add_argument('--bvh_dir', '-orig', required=True,
                                    help="Path where original motion files (in BVH format) are stored")
-    parser.add_argument('--dest_dir', '-dest', default="./processed/",
+    parser.add_argument('--dest_dir', '-dest', required=True,
                                    help="Path where extracted motion features will be stored")
     parser.add_argument('--pipeline_dir', '-pipe', default="./utils/",
                         help="Path where the motion data processing pipeline will be stored")

data_processing/features2bvh.py

@@ -15,12 +15,12 @@
 from pymo.viz_tools import *
 from pymo.writers import *
 
-import joblib as jl
+from argparse import ArgumentParser
 
-# load data pipeline
-pipeline = jl.load('processed/data_pipe.sav')
+import joblib as jl
 
 def feat2bvh(feat_file, bvh_file):
+
     features = np.load(feat_file)['clips']
     print(features.shape)
 
@@ -34,4 +34,22 @@ def feat2bvh(feat_file, bvh_file):
     with open(bvh_file,'w') as f:
         writer.write(bvh_data[0], f)
 
-feat2bvh("processed/NaturalTalking_002.npz", 'processed/converted2.bvh')
+if __name__ == '__main__':
+
+    # Setup parameter parser
+    parser = ArgumentParser(add_help=False)
+    parser.add_argument('--feat_dir', '-feat', required=True,
+                                   help="Path where motion features are stored")
+    parser.add_argument('--bvh_dir', '-bvh', required=True,
+                                   help="Path where produced motion files (in BVH format) will be stored")
+    parser.add_argument('--pipeline_dir', '-pipe', default="./utils/",
+                        help="Path where the motion data processing pipeline is be stored")
+
+    params = parser.parse_args()
+
+
+    # load data pipeline
+    pipeline = jl.load('processed/data_pipe.sav')
+    
+    # convert a file
+    feat2bvh(params.feat_dir, params.bvh_dir)

data_processing/utils/data_pipe.sav

@@ -10,6 +10,10 @@
 from learn_ae_n_encode_dataset import create_nn, prepare_motion_data
 from config import args
 
+import numpy as np
+
+import sys
+
 if __name__ == '__main__':
     # Make sure that the two mandatory arguments are provided
     if args.input_file is None or args.output_file is None:

motion_repr_learning/ae/decode.py

@@ -3,6 +3,9 @@
 """
 
 from os.path import join
+import sys
+
+sys.path.append('.')
 
 import numpy as np
 import tensorflow as tf