es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

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?

Tags:  , , ,

Answer

  1. Avatar for 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 de keras.models.load_model. El método load_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:

    import tensorflow as tf
    
    model_path = 'ruta/hacia/carpeta_de_modelo_tf2'
    model = tf.keras.models.load_model(model_path)
    

    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.

Comments are closed.