En Python, ¿cómo obtener el tamaño de imágenes del modelo de TensorFlow2 entrenado?

14 Jul, 2023 Programación 0

Estoy cargando un modelo usando:

m = tf.saved_model.load(str(model_path))

Quiero poder obtener la información del tamaño de las imágenes en las que el modelo fue entrenado para poder redimensionar las nuevas imágenes en las que deseo hacer inferencias.

Sé que con un modelo de Keras puedo usar:

shape_0 = m.layers[0].output_shape
input_height = shape_0[1]
input_width = shape_0[2]

para obtener la altura y el ancho de entrada de las imágenes de entrenamiento.

¿Existen comandos similares para obtener estos valores de un modelo en TensorFlow 2?

saved_model = tf.saved_model.load(str(model_path)) signature = list(saved_model.signatures.values())[0] input_shape = signature.inputs['input_name'].shape.as_list()[1:3] # reemplaza 'input_name' con el nombre de tu capa de entrada print(input_shape)

Answer

davy.ai

July 14, 2023 at 10:51 am
Sí, existen comandos similares para obtener la forma de entrada de un modelo de TensorFlow 2. Aquí tienes un ejemplo de cómo obtener la forma de entrada de un modelo de TensorFlow 2:
```
saved_model = tf.saved_model.load(str(model_path))
signature = list(saved_model.signatures.values())[0]
input_shape = signature.inputs['input_name'].shape.as_list()[1:3] # reemplaza 'input_name' con el nombre de tu capa de entrada
print(input_shape)
```
Esto te dará la forma de entrada (altura y anchura) de la capa de entrada del modelo. Luego, puedes utilizar esta información para redimensionar las nuevas imágenes que deseas inferir.

Comments are closed.

es.davy.ai

¿Tienes una pregunta?

En Python, ¿cómo obtener el tamaño de imágenes del modelo de TensorFlow2 entrenado?

Answer

davy.ai