Actualización del programa de Python para utilizar modelos de TensorFlow 2 en Keras 2.4.
Estoy usando keras 2.4 y tengo el siguiente código:
importar modelos keras
ruta_modelo = path\tf1_model.model
ruta_json = path\tf1_model.json
m = keras.models.load_model(str(ruta_modelo))
Esto funciona perfectamente bien. Pero quiero actualizar para usar modelos tensorflow 2. Mi investigación indica que para la ruta del modelo, debo apuntar a donde está ahora la carpeta del modelo TF2 que contiene el archivo .pb, la carpeta de activos y la carpeta de variables en lugar de cualquier archivo dado como el archivo .pb.
Cuando intento esto, obtengo el error Cannot find C:\Users\722208\Desktop\Models\WPC_Ox_TF2_112421_Segnet_1_serving_model, aunque sé que puede ver la carpeta, porque intenté apuntar al archivo .pb y el programa lo ve pero falla al cargarlo.
¿Estoy haciendo todo mal?
davy.ai
Parece que estás intentando cargar un modelo de TensorFlow 2 usando Keras 2.4. Si bien es posible ejecutar modelos de TensorFlow 2 usando Keras, se recomienda utilizar la API de TensorFlow 2 en su lugar.
Para cargar un modelo de TensorFlow 2, puedes utilizar el método
tensorflow.keras.models.load_model
en lugar dekeras.models.load_model
. El métodoload_model
puede tomar la ruta ya sea de la carpeta del modelo guardado o del archivo .h5 directamente.Aquí tienes un ejemplo de cómo cargar un modelo de TensorFlow 2 usando
tensorflow.keras
:Asegúrate de que la variable
model_path
apunte a la carpeta que contiene los archivos del modelo guardado (archivo protobuf, carpeta de activos y carpeta de variables).Si todavía tienes dificultades para cargar el modelo, intenta verificar de nuevo la ruta hacia la carpeta del modelo y asegurarte de que los archivos del modelo guardado tengan el formato correcto. También puede ser útil consultar la documentación de TensorFlow para obtener más información sobre cómo guardar y cargar modelos.