Esta página se tradujo automáticamente con la API de traducción de Google Cloud.
Algunas páginas se pueden leer mejor en su totalidad.
TensorFlow.js es una poderosa biblioteca de JavaScript para entrenar e implementar modelos de aprendizaje automático. Se puede usar con otros marcos de ML, como TensorFlow Lite, para ejecutar inferencias en Node.js.
En esta publicación, le mostraremos cómo usar TensorFlow.js con otros marcos de ML en Node.js. También brindaremos algunos consejos sobre cómo aprovechar TensorFlow.js al máximo.
TensorFlow.js se puede usar con TensorFlow Lite para ejecutar inferencias en Node.js. TensorFlow Lite es un marco ligero para ejecutar modelos de aprendizaje automático en dispositivos móviles.
Para usar TensorFlow.js con TensorFlow Lite, deberá instalar los enlaces de TensorFlow Lite Node.js. Puedes hacer esto con npm:
npm install @tensorflow/tfjs-node
Una vez que se instalan los enlaces, puede usar el módulo tensorflow.js
para cargar un modelo de TensorFlow Lite y ejecutar la inferencia en él.
Este es un ejemplo de cómo usar TensorFlow.js con TensorFlow Lite:
const tf = require('@tensorflow/tfjs-node');
// Load a TensorFlow Lite model.
const model = await tf.loadLiteModel('https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_1.0_224/model.json');
// Run inference on a TensorFlow Lite model.
const output = model.predict(tf.zeros([1, 224, 224, 3]));
// Print the output of the model.
console.log(output);
TensorFlow.js también se puede usar con otros marcos de ML, como Keras, para ejecutar inferencias en Node.js.
Para usar TensorFlow.js con Keras, deberá instalar los enlaces TensorFlow.js Node.js. Puedes hacer esto con npm:
npm install @tensorflow/tfjs
Una vez que se instalan los enlaces, puede usar el módulo tf
para cargar un modelo de Keras y ejecutar la inferencia en él.
Aquí hay un ejemplo de cómo usar TensorFlow.js con Keras:
const tf = require('@tensorflow/tfjs');
// Load a Keras model.
const model = await tf.loadModel('https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_1.0_224/model.json');
// Run inference on a Keras model.
const output = model.predict(tf.zeros([1, 224, 224, 3]));
// Print the output of the model.
console.log(output);
Estos son algunos consejos para usar TensorFlow.js: