Moved smoothing to the gesture space

(from the encoding space)

Author: Svito-zar

motion_repr_learning/ae/decode.py

@@ -14,6 +14,18 @@
 
 import sys
 
+from scipy.signal import savgol_filter
+
+
+def smoothing(motion):
+
+    smoothed = [savgol_filter(motion[:,i], 9, 3) for i in range(motion.shape[1])]
+
+    new_motion = np.array(smoothed).transpose()
+
+    return new_motion
+
+
 if __name__ == '__main__':
     # Make sure that the two mandatory arguments are provided
     if args.input_file is None or args.output_file is None:
@@ -39,6 +51,9 @@
     # Decode it
     decoding = tr.decode(nn, encoding)
 
+    # Smoothing
+    decoding = smoothing(decoding)
+
     print(decoding.shape)
 
     np.save(args.output_file, decoding)

predict.py

@@ -7,17 +7,6 @@
 import sys
 from keras.models import load_model
 import numpy as np
-from scipy.signal import savgol_filter
-
-
-def smoothing(motion):
-
-    smoothed = [savgol_filter(motion[:,i], 9, 3) for i in range(motion.shape[1])]
-
-    new_motion = np.array(smoothed).transpose()
-
-    return new_motion
-
 
 def predict(model_name, input_file, output_file):
     """ Predict human gesture based on the speech
@@ -35,10 +24,8 @@ def predict(model_name, input_file, output_file):
 
     predicted = np.array(model.predict(audio))
 
-    smoothed = smoothing(predicted)
-
     print("Encoding shape is: ", predicted.shape)
-    np.savetxt(output_file, smoothed)
+    np.savetxt(output_file, predicted)
 
 
 if __name__ == "__main__":