Merge pull request #312 from JdeRobot/issue-311

Updated keras brain with new TF version
JdeRobot · Feb 4, 2022 · 49c3f55 · 49c3f55
2 parents ee35ffa + 156f056
commit 49c3f55
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/behavior_metrics/brains/f1/brain_f1_keras.py b/behavior_metrics/brains/f1/brain_f1_keras.py
@@ -9,9 +9,8 @@
 
 """
 
+import tensorflow as tf
 import numpy as np
-
-import keras
 import cv2
 import time
 import os
@@ -57,7 +56,8 @@ def __init__(self, sensors, actuators, model=None, handler=None, config=None):
             if not path.exists(PRETRAINED_MODELS + model):
                 print("File " + model + " cannot be found in " + PRETRAINED_MODELS)
 
-            self.net = keras.models.load_model(PRETRAINED_MODELS + model)
+            self.net = tf.keras.models.load_model(PRETRAINED_MODELS + model)
+            print(self.net.summary())
         else:
             print("** Brain not loaded **")
             print("- Models path: " + PRETRAINED_MODELS)