Cómo entrenar tus modelos personalizados con Ultralytics HUB

A medida que nos esforzamos constantemente por la innovación impulsada por IA, únete a nosotros mientras analizamos más de cerca Ultralytics HUB, una plataforma diseñada para simplificar el entrenamiento de modelos personalizados Ultralytics YOLO. Tanto si eres un desarrollador experimentado como si eres principiante, Ultralytics HUB ofrece una experiencia fluida para crear y gestionar modelos de visión artificial. Exploremos cómo entrenar tus modelos YOLO personalizados en solo unos clics.

Link to this section¿Qué es Ultralytics HUB?#

Ultralytics HUB es una plataforma integral diseñada para gestionar tus conjuntos de datos, proyectos y modelos, lo que facilita entrenar y desplegar modelos de aprendizaje automático.

Se integra con varias plataformas, admite actualizaciones en tiempo real a través de su aplicación móvil (disponible tanto en Android como en iOS), y evoluciona continuamente con nuevas funciones. Para obtener más información técnica y un flujo de trabajo paso a paso, consulta nuestro artículo en Medium sobre cómo entrenar modelos de visión artificial en la nube.

Link to this sectionPrimeros pasos con Ultralytics HUB#

Link to this sectionInterfaz fácil de usar#

Lo primero que notarás en Ultralytics HUB es su interfaz intuitiva. La página de inicio proporciona un fácil acceso a conjuntos de datos, proyectos, modelos e integraciones. Aunque todavía está en fase beta, la plataforma ya cuenta con una gama de funciones potentes.

Link to this sectionConjuntos de datos y modelos#

Ultralytics HUB viene precargado con conjuntos de datos estándar como VOC, COCO y Simpsons, que se utilizan habitualmente para evaluaciones comparativas. Estos conjuntos de datos son sustanciales, ya que COCO cuenta con 80 clases y 140 000 imágenes. Sin embargo, la plataforma también admite conjuntos de datos más pequeños, lo que la hace ideal para usuarios de todos los niveles. Puedes subir tu conjunto de datos personalizado o conectarte con herramientas externas como Roboflow para la anotación de imágenes y la preparación del conjunto de datos.

Link to this sectionCreación de un proyecto#

Crear un nuevo proyecto en Ultralytics HUB es increíblemente sencillo. Aquí tienes una guía paso a paso para empezar:

1. Crea un proyecto: Empieza por ponerle un nombre a tu proyecto. Para este ejemplo, llamémoslo "Detección de objetos".
2. Elige un conjunto de datos: Selecciona un conjunto de datos de las opciones disponibles. Para simplificar, usaremos el conjunto de datos de Simpsons con 14 000 imágenes.
3. Selecciona un modelo: Elige un modelo YOLO. Ultralytics HUB admite varios modelos, incluidos YOLOv5 y YOLOv8. Para este ejemplo, utilizaremos el modelo YOLOv8 nano.
4. Ajusta los hiperparámetros: Ajusta la configuración, como el número de épocas, el tamaño de la imagen y el tamaño del lote. También puedes seleccionar si deseas usar una GPU o una CPU y configurar las opciones de almacenamiento en caché.

Link to this sectionEntrenamiento del modelo#

Una vez configurado tu proyecto, entrenar el modelo está a solo un clic de distancia. Ultralytics HUB ofrece múltiples opciones de entrenamiento:

• Entrenamiento local: Instala Ultralytics en tu máquina local y ejecuta el script de entrenamiento.
• Google Colab: Abre un cuaderno de Google Colab preconfigurado con el código necesario. Esta opción no requiere programación y se ejecuta sin problemas en la nube.
• HUB Cloud: El entrenamiento en la nube de Ultralytics HUB ofrece una solución sin código para entrenar modelos YOLO, ideal para personas que no programan y propietarios de empresas. El flujo de trabajo incluye la carga del conjunto de datos, la selección del modelo y la configuración de la instancia en la nube, lo que facilita el ajuste de modelos preentrenados y su exportación para diversas aplicaciones.

Para entrenar el modelo en Google Colab:

1. Instala Ultralytics: Ejecuta el comando para instalar Ultralytics en el cuaderno.
2. Configuración y clave de API: Completa la configuración e introduce tu clave de API.
3. Inicia el entrenamiento: Ejecuta los comandos de entrenamiento y tu modelo comenzará a entrenarse.

Fig 1. Nicolai Nielsen explicando cómo entrenar modelos en Ultralytics HUB.

Link to this sectionSeguimiento del progreso del entrenamiento#

Ultralytics HUB proporciona un seguimiento en tiempo real del progreso del entrenamiento de tu modelo. Puedes supervisar métricas clave como la precisión, la exhaustividad (recall) y las funciones de pérdida. La plataforma también visualiza los datos de entrenamiento, lo que te permite ver cómo mejora tu modelo con el tiempo.

Link to this sectionFunciones avanzadas e integraciones#

Link to this sectionEntrenamiento en la nube e integración con Roboflow#

Ultralytics HUB ha introducido el entrenamiento en la nube, permitiendo a los usuarios entrenar modelos directamente en la nube sin ninguna configuración local. La plataforma también se integra con Roboflow para anotar imágenes y preparar conjuntos de datos. Esta integración agiliza el flujo de trabajo desde la preparación de datos hasta el entrenamiento del modelo.

Link to this sectionAplicación móvil#

La aplicación Ultralytics HUB lleva la detección de objetos en tiempo real a tu dispositivo móvil. Usando modelos preentrenados, la aplicación puede detectar objetos del conjunto de datos COCO con gran precisión, funcionando a 30 fotogramas por segundo en un iPhone 14 Pro. Esta función es perfecta para probar modelos en escenarios del mundo real y demostrar las capacidades de tus modelos entrenados.

Link to this sectionConclusión#

Ultralytics HUB es un punto de inflexión en el mundo de la visión artificial, haciendo que sea más fácil que nunca entrenar modelos YOLO personalizados. Su interfaz fácil de usar, sus funciones robustas y sus integraciones fluidas lo posicionan como una herramienta imprescindible para desarrolladores e investigadores por igual. Tanto si buscas entrenar modelos de forma local, en la nube o sobre la marcha con la aplicación móvil, Ultralytics HUB tiene todo lo que necesitas.

Así que, ¿por qué esperar? ¡Sumérgete en Ultralytics HUB hoy mismo y desbloquea el potencial de tus proyectos de aprendizaje automático con solo unos clics!