Merge pull request #19 from akapoorcern/master

Added some functionalities
cms-egamma · Jul 25, 2021 · 1317697 · 1317697
2 parents 197bcfb + 2a9d996
commit 1317697
Show file tree

Hide file tree

Showing 3 changed files with 2,198 additions and 452 deletions.
diff --git a/Tools/convert_h5Topb.py b/Tools/convert_h5Topb.py
@@ -0,0 +1,11 @@
+import numpy as np
+import tensorflow as tf
+from tensorflow import keras
+import sys
+
+modelloc=str(sys.argv[1])
+pbmodelloc=str(sys.argv[2])
+model = keras.models.load_model(modelloc)
+
+import cmsml
+cmsml.tensorflow.save_graph(pbmodelloc, model, variables_to_constants=True)