¿Cómo cargar un modelo entrenado en formato TFLite en un navegador?

Answer

1. 

   davy.ai

   July 19, 2023 at 6:07 am

   Para cargar un modelo tflite en un navegador, específicamente utilizando Tensorflow.js, puedes seguir estos pasos:

   1. Descarga el modelo tflite del repositorio de Github especificado.
   2. Instala Tensorflow.js usando el comando: npm install @tensorflow/tfjs
   3. Convierte el modelo tflite a un formato compatible con TensorFlow.js utilizando el comando:
      python -m tf2onnx.convert --saved-model ruta/al/tflite --output modelo.onnx --opset 13
      Esto generará un archivo .onnx que puede ser cargado por Tensorflow.js.
   4. Carga el modelo tflite utilizando:

   import * as tf from '@tensorflow/tfjs';
   
   const modelo = await tf.loadGraphModel('ruta/al/modelo.onnx');
   

   1. Ahora puedes utilizar el modelo cargado para hacer predicciones o realizar inferencias en nuevos datos.

   Nota: Para utilizar este enfoque, deberás configurar tu aplicación de Tensorflow.js con el backend de TensorFlow.js, como se describe en la documentación de Tensorflow.js.

Comments are closed.