Yolo Vision Shenzhen
Shenzhen
Únete ahora

Formación personalizada de modelos Ultralytics YOLO en Lightning AI

Abirami Vina

3 minutos de lectura

13 de enero de 2025

Descubra cómo Lightning AI, presentada en YOLO Vision 2024, simplifica el desarrollo escalable de la IA de visión con un entrenamiento, despliegue y colaboración de modelos más rápidos.

Tanto si eres un desarrollador de IA experimentado como si acabas de empezar a explorar la IA de visión, disponer de un entorno fiable para jugar y experimentar con modelos de visión por ordenador como Ultralytics YOLO11 es fundamental. Un entorno hace referencia a las herramientas, los recursos y la infraestructura necesarios para diseñar, probar y desplegar modelos de IA de forma eficiente. 

Si bien varias plataformas en línea ofrecen diferentes herramientas de IA, muchas no proporcionan un entorno unificado para todo el ciclo de vida de la IA, desde la preparación de datos hasta la implementación del modelo. Aquí es donde Lightning AI, una plataforma todo en uno para el desarrollo de la IA, interviene para agilizar el proceso desde la preparación de los datos hasta la implementación.

La importancia de facilitar el desarrollo de la IA se puso de manifiesto en YOLO Vision 2024 (YV24), un evento híbrido anual organizado por Ultralytics que se centró en los avances de la IA y la visión por ordenador. Luca Antiga, director de tecnología de Lightning AI, ofreció una charla magistral titulada"Going YOLO on Lightning Studios", en la que explicó cómo entrenar los modelosYOLO Ultralytics de forma rápida, fluida y sin involucrarse en complejidades técnicas mediante Lightning AI.

En este artículo, nos sumergiremos en los puntos clave de la charla de Luca, cubriendo todo, desde aplicaciones de visión por ordenador del mundo real hasta demostraciones en vivo sobre la formación y el despliegue de modelos Ultralytics YOLO con Lightning AI. Comencemos.

Uso de Lightning AI y Ultralytics YOLO para simplificar el desarrollo de la IA

Luca comenzó su discurso compartiendo sus ideas y su apreciación de la influencia de los modelos YOLO en diversas industrias. Destacó cómo los modelos YOLO pueden aplicarse en sectores como la fabricación y la agricultura. Aprecio el impacto que YOLO ha tenido en la comunidad de constructores, personas que necesitan resolver problemas reales y prácticos.

Conectando esto al creciente interés en el entrenamiento de la IA, presentó Lightning AI, una plataforma diseñada para hacer que el desarrollo de modelos de IA sea más rápido, más simple y más accesible para todos. Es especialmente útil para apoyar los avances iterativos en la IA, ayudando a los desarrolladores a refinar y mejorar los modelos.

Fig. 1. Luca Antiga presentando de forma remota Lightning Studios en YV24.

También señaló que Lightning AI es similar a PyTorch Lightning, un marco que simplifica el proceso de entrenamiento de modelos de IA. Sin embargo, en lo que difiere es en que Lightning AI es una plataforma más completa que proporciona un conjunto más amplio de herramientas y capacidades para todo el proceso de desarrollo de IA, no solo para el entrenamiento de modelos de IA. 

Un componente vital de Lightning AI es Lightning Studios, que ofrece un espacio de trabajo intuitivo para diseñar, entrenar y desplegar modelos de IA, haciendo que todo el flujo de trabajo sea fluido y eficiente. Puede pensar en Lightning Studios como un entorno de desarrollo reproducible para la IA que se ejecuta en la nube. Por ejemplo, ofrece un entorno similar a Jupyter Notebook que puede ser duplicado y compartido con otro desarrollador, ayudando a mejorar la colaboración. 

A continuación, Luca se explayó sobre las ventajas de Lightning Studios: "Replicar tu entorno ya no es un problema. Si necesitas cambiar de una máquina CPU [Unidad Central de Procesamiento] a una máquina GPU [Unidad de Procesamiento Gráfico] o lanzar un entrenamiento a través de mil máquinas, tu entorno será persistente."

Configurando Lightning Studios para el entrenamiento y el desarrollo

A continuación, Luca demostró lo rápido que se puede empezar a trabajar con Lightning Studios. Con unos pocos clics, puedes abrir un nuevo estudio y acceder a herramientas y entornos como Jupyter Notebooks y VS Code, todo configurado y listo para codificar. Mostró lo fácil que es cambiar de una máquina a otra. Si la tarea en la que estás trabajando requiere más potencia, puedes cambiar fácilmente de una CPU a una GPU más potente. La GPU permanecerá activa sólo mientras se esté utilizando; de lo contrario, entrará en modo de reposo, ahorrando tus créditos.

Luca también mencionó los beneficios de usar Studio Templates. Son entornos de codificación de IA prefabricados por la comunidad, y puede usarlos sin tener que configurar nada. Configurar un entorno para proyectos de IA puede llevar mucho tiempo, y Studio Templates puede ayudar a aumentar la productividad. Estos entornos vienen precargados con todo lo necesario para proyectos de IA, como dependencias instaladas, pesos de modelos, datos, código, etc.

Fig 2. Luca explicando qué son las plantillas de Studio.

Entrenamiento de los modelosYOLO Ultralytics en Lightning Studios

A continuación, Luca pasó a la demostración en vivo, destacando cómo se puede utilizar Lightning Studio para entrenar modelosUltralytics YOLO . Abrió una plantilla de Studio, que ya tenía instaladas todas las dependencias, y puso en marcha una máquina con cuatro GPU para acelerar el proceso de entrenamiento. Con respecto a los datos, dijo que se puede optar por almacenar los datos directamente en la máquina o transmitirlos desde la nube, lo que hace que el proceso de formación sea más rápido y eficiente.

En pocos segundos, la máquina estaba lista y Luca inició rápidamente la sesión de entrenamiento. Durante la demostración, un problema menor hizo que la máquina se detuviera inesperadamente, pero Lightning Studios reanudó sin problemas desde donde lo había dejado, asegurándose de que no se perdiera ningún progreso. Luca señaló cómo esta fiabilidad favorece flujos de trabajo fluidos, incluso ante interrupciones inesperadas.

Continuando con la demostración, mostró lo fácil que es monitorizar el progreso del entrenamiento utilizando TensorBoard, una herramienta para visualizar las métricas de aprendizaje automático en tiempo real. Lightning Studio lo simplifica aún más generando automáticamente URLs que permiten a usted o a sus compañeros de equipo en el mismo espacio de trabajo acceder a las vistas de TensorBoard sin ninguna configuración adicional. Esto agiliza la colaboración y mantiene a todos al tanto. 

Fig. 3. Diagrama de flujo del entrenamiento de los modelosYOLO Ultralytics en Lightning Studios. Imagen del autor.

Despliegue de modelosYOLO Ultralytics con Lit Serve

Después de la demostración, Luca cambió el enfoque de la charla a un nuevo proyecto, LitServe, lanzado recientemente por Lightning AI. LitServe simplifica el proceso de tomar un modelo entrenado y convertirlo en un servicio escalable que otros puedan utilizar, eliminando la necesidad de complejas canalizaciones de implementación. Está diseñado para gestionar todo, desde el empaquetado del modelo hasta su implementación con el mínimo esfuerzo.

Para mostrarlo en tiempo real, Luca ofreció a la audiencia una demostración rápida utilizando un Ultralytics YOLOv8 previamente entrenado. Fue capaz de crear una API sencilla para gestionar las solicitudes entrantes y devolver predicciones de imágenes en unos segundos. Esto significa que cualquiera puede hacer ping a esta API con una imagen y recibir resultados para tareas de visión por ordenador como la detección de objetos casi al instante. Entre bastidores, el modeloYOLOv8 Ultralytics se despliega como un servicio que gestiona eficazmente las solicitudes, procesa las imágenes y proporciona predicciones con una latencia mínima.

Fig 4. Luca mostrando LitServe de Lightning AI durante YV24.

Ejecutó una inferencia sobre una imagen de pizza, y Ultralytics YOLOv8 identificó con éxito objetos como la pizza, una cuchara y una mesa de comedor. Explicó que, aunque la primera solicitud tarda un poco más debido al "arranque en frío", las siguientes son mucho más rápidas una vez que el sistema se calienta.

Luca luego preguntó: '¿Qué pasa si quiero exponer esto al mundo exterior?' Explicó cómo el complemento API Builder simplifica la conversión de su modelo en un servicio en vivo, listo para producción. Con características como dominios personalizados, seguridad adicional e integración perfecta, puede hacer que su modelo sea accesible para cualquier persona fácilmente.

Ventajas clave de usar Lightning Studios

Para concluir su charla, Luca habló de la escalabilidad y la flexibilidad de Lightning Studio para el desarrollo de la IA. Mencionó cómo la plataforma puede entrenar modelos en múltiples máquinas, escalando hasta 10.000 nodos, con un entrenamiento tolerante a fallos que se reanuda automáticamente después de cualquier interrupción. 

Por ejemplo, si un trabajo de entrenamiento en un clúster de GPU se interrumpe debido a un problema de hardware o a un reinicio del servidor, Lightning Studios se asegura de que el proceso se reanude exactamente donde se quedó. Esto lo hace ideal para proyectos de IA a gran escala, como el entrenamiento de modelos de aprendizaje profundo en conjuntos de datos masivos como ImageNet o COCO.

Aquí hay algunos otros beneficios clave de Lightning Studios de los que habló Luca:

  • Créditos de GPU mensuales gratuitos: Los usuarios reciben 15 créditos de GPU gratuitos cada mes, que se rellenan automáticamente, lo que garantiza que puedas experimentar y desarrollar sin costes añadidos.
  • Colaboración mejorada: Los espacios de equipo compartidos y los entornos reproducibles de Lightning Studio permiten a los miembros del equipo trabajar juntos sin problemas, garantizando la coherencia y la eficiencia en todos los proyectos.
  • Opciones de instancia flexibles: Ofrece la flexibilidad de elegir entre instancias interrumpibles y no interrumpibles, lo que permite a los usuarios ahorrar costes en máquinas GPU con opciones interrumpibles.
  • Integración con herramientas existentes: La plataforma se integra con herramientas de desarrollo remoto como SSH (Secure Socket Shell) y VS Code, proporcionando flexibilidad para trabajar localmente o en la nube.

Conclusiones clave

La ponencia de Luca en YV24 puso de relieve cómo la IA, combinada con herramientas como los modelosYOLO Ultralytics y Lightning AI, está cambiando la forma en que resolvemos los problemas del mundo real. Facilitan a los desarrolladores la formación y el despliegue de modelos diseñados para abordar problemas específicos en diversos sectores.

Ilustró cómo Lightning Studios hace que todo el proceso de desarrollo sea más rápido y accesible, permitiendo a los desarrolladores crear soluciones potentes con facilidad. En el núcleo de plataformas de vanguardia como Lightning AI, los modelos de visión por ordenador están transformando la forma en que las soluciones de IA afrontan los retos. En particular, con el último modelo Ultralytics YOLO11 , los desarrolladores pueden crear soluciones que tengan un impacto significativo.

Únase a nuestra comunidad para mantenerse al día sobre la IA y sus usos prácticos. Consulte nuestro repositorio de GitHub para explorar innovaciones en sectores como la IA en la conducción autónoma y la visión artificial en la atención médica.

¡Construyamos juntos el futuro
de la IA!

Comience su viaje con el futuro del aprendizaje automático

Comienza gratis