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.

Descargue solo una parte específica de la biblioteca de TensorFlow.

Tengo un código de Aprendizaje Profundo para Detectar Objetos. Lo que hice fue correr el código en Google Colab y exporté el modelo para usarlo localmente. Ahora, para ejecutar el modelo, debo volver a instalar todo el paquete de Tensorflow, lo cual es demasiado pesado para mi sistema.

Quisiera preguntar si hay alguna manera de descargar y ejecutar solo partes específicas de la biblioteca de Tensorflow.

Estoy usando Tensorflow solo en 2 lugares de mi código y tengo que instalar toda la biblioteca de Tensorflow para ello.

  • Aquí es donde estoy cargando el modelo.
`detect_fn = tf.saved_model.load(PATH_TO_SAVED_MODEL)`
  • Este es el segundo lugar donde utilizo Tensorflow.

    input_tensor = tf.convert_to_tensor(image_rgb)

Solo necesito estas 2 funciones de la Biblioteca de Tensorflow y no toda la biblioteca… Gracias anticipadamente.

Tags:  , , , ,

Answer

  1. Avatar for davy.ai

    Sí, es posible descargar y ejecutar solo partes específicas de la biblioteca de Tensorflow. Una forma de lograrlo es utilizando la estructura modular de Tensorflow e importando solo los módulos o funciones necesarios. Por ejemplo, en su caso, primero puede verificar qué módulos específicos se requieren para las funciones que está utilizando (“detect_fn” y “convert_to_tensor”). Luego, puede utilizar el siguiente código para instalar solo esos módulos específicos:

    !pip install tensorflow-<nombre_del_módulo>
    

    Reemplace <nombre_del_módulo> con el nombre del módulo específico requerido por su código. Por ejemplo, si “detect_fn” requiere el módulo “saved_model”, puede instalarlo usando:

    !pip install tensorflow-saved_model
    

    De manera similar, si “convert_to_tensor” requiere el módulo “compat”, puede instalarlo usando:

    !pip install tensorflow-compat
    

    Al instalar solo los módulos requeridos, puede evitar instalar toda la biblioteca de Tensorflow, lo cual puede ser pesado para su sistema.</nombre_del_módulo></nombre_del_módulo>

Comments are closed.