Merge pull request #73 from ANaaim/Optimise_MarkerAugmentation

Clear keras session to avoid time increase when doing batch processing

Pose2Sim/markerAugmentation.py

@@ -31,6 +31,7 @@ from Pose2Sim.MarkerAugmenter.utils import TRC2numpy
 import json
 import glob
 import logging
+import keras
 
 
 ## AUTHORSHIP INFORMATION
@@ -183,6 +184,7 @@ def augmentTRC(config_dict):
             model = tf.keras.models.model_from_json(pretrainedModel_json)
             model.load_weights(os.path.join(augmenterModelDir, "weights.h5"))  
             outputs = model.predict(inputs)
+            keras.backend.clear_session()
 
             # %% Post-process outputs.
             # Step 1: Reshape if necessary (eg, LSTM)