fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #73 from ANaaim/Optimise_MarkerAugmentation

Browse Source 
Clear keras session to avoid time increase when doing batch processing
This commit is contained in:
David PAGNON 2024-02-27 21:30:06 +01:00 committed by GitHub
commit 62534a9229
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Pose2Sim/markerAugmentation.py
View File

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