Crea y entrena una IA desde cero con TensorFlow.js y Keras para navegadores web

Hace unos días, mientras compartíamos ideas en la comunidad de WhatsApp de Realidad Aumentada «Empezando Desde Cero», surgió una idea fascinante: ¿cómo podríamos crear una Inteligencia Artificial que pudiera ser entrenada y luego implementada en un entorno de Realidad Aumentada?
Inspirado por esta pregunta, decidí embarcarme en este proyecto y hoy quiero compartir el resultado. A continuación, les mostraré cómo construí esta aplicación de inteligencia artificial utilizando TensorFlow.js y Keras, y les contaré todo lo que aprendí en el proceso.
Antes en el blog ya habíamos realizado apps de realidad aumentada utilizando inteligencia artificial:
- Realidad Aumentada en Javascript con inteligencia artificial
- Inteligencia artificial en una app de realidad aumentada
- Utiliza el micrófono de Android en Realidad Aumentada
¿Cómo funciona la aplicación?
La aplicación, construida con TensorFlow.js, Keras, HTML, CSS y JavaScript, te permite entrenar a una IA para que reconozca cualquier objeto.
Pasos para entrenar el modelo:
- Activa tu cámara web.
- Agrega un objeto: Dale un nombre al objeto que quieres reconocer y haz clic en «Añadir Objeto».
- Captura las imágenes: Comienza a tomar fotos del objeto. Consejo: para un mejor entrenamiento, captura el objeto desde diferentes ángulos, con distinta iluminación y sobre fondos variados. Cuantas más fotos, mejor será el resultado.
- Entrena el modelo: Haz clic en «Entrenar Modelo» y espera a que aparezca el mensaje de «Entrenamiento completado. ¡Puedes iniciar el reconocimiento!».
- Inicia el reconocimiento: Haz clic en «Iniciar Reconocimiento» y coloca el objeto delante de la cámara para ver cómo la IA lo identifica.
Funcionalidades clave
- Reconocimiento de múltiples objetos: Puedes entrenar la IA para que reconozca uno o varios objetos.
- Guardar y continuar: El modelo entrenado se descarga en un archivo .zip. Más tarde, puedes volver a subir este archivo para continuar el entrenamiento y agregar más objetos a tu modelo.
- Modelos portátiles: El archivo .zip contiene todo lo necesario para que tu modelo entrenado pueda ser usado en cualquier momento. Ten en cuenta que el tamaño del archivo aumentará a medida que agregues más objetos e imágenes.
Este es el resultado de la version 1.0.0, pruébala en tu pc online aquí: IA de reconocimiento de objetos: TensorFlow.js y Keras
Próximamente, publicaré un tutorial completo y detallado. Estaré actualizando esta página con los enlaces a cada una de las partes de la serie, ¡así que vuelve a visitarla pronto!
Segunda parte de este tutorial: Inteligencia Artificial en la Web: TensorFlow.js y Keras
Tercera parte de este tutorial: App con TensorFlow.js y Keras – IA para reconocer objetos
Cuarta parte de este tutorial: App con TensorFlow.js y Keras – HTML y CSS explicado
Mira el video actualizado de la versión 2.0