Cómo entrenar sus modelos personalizados con Ultralytics HUB

En nuestra búsqueda constante de innovación impulsada por la IA, únase a nosotros para analizar más de cerca Ultralytics HUB, una plataforma diseñada para simplificar el entrenamiento de modelos Ultralytics YOLO personalizados. Tanto si es un desarrollador experimentado como si es un principiante, Ultralytics HUB ofrece una experiencia perfecta para crear y gestionar modelos de visión artificial. Exploremos cómo entrenar sus modelos YOLO personalizados con tan solo unos clics.

¿Qué es Ultralytics HUB?

Ultralytics HUB es una plataforma integral diseñada para gestionar sus conjuntos de datos, proyectos y modelos, lo que facilita el entrenamiento y el despliegue de 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 está en continua evolución con nuevas funciones. Para obtener información técnica más detallada y un flujo de trabajo paso a paso, consulte nuestro artículo de Medium sobre cómo entrenar modelos de visión artificial en la nube.

Primeros pasos con Ultralytics HUB

Interfaz fácil de usar

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

Conjuntos de datos y modelos

Ultralytics HUB viene precargado con conjuntos de datos estándar como VOC, COCO y Simpsons, que se utilizan habitualmente para la evaluación comparativa. Estos conjuntos de datos son importantes, con COCO 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. Puede cargar su conjunto de datos personalizado o conectarse con herramientas externas como Roboflow para la anotación de imágenes y la preparación de conjuntos de datos.

Creación de un proyecto

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

1. Crear un proyecto: Comience por nombrar su proyecto. Para este ejemplo, llamémoslo "Detección de objetos".
   ‍
2. Elija un conjunto de datos: Seleccione un conjunto de datos de las opciones disponibles. Para simplificar, utilizaremos el conjunto de datos de los Simpsons con 14.000 imágenes.
   ‍
3. Seleccione un modelo: Elija un modelo YOLO. Ultralytics HUB admite varios modelos, incluidos YOLOv5 y YOLOv8. Para este ejemplo, utilizaremos el modelo YOLOv8 nano.
   ‍
4. Ajustar hiperparámetros: Ajuste la configuración, como el número de épocas, el tamaño de la imagen y el tamaño del lote. También puede seleccionar si desea utilizar una GPU o CPU y configurar las opciones de almacenamiento en caché.

Entrenamiento del modelo

Una vez que tu proyecto está configurado, 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 codificación y se ejecuta a la perfección en la nube.
  ‍
• HUB Cloud: Ultralytics HUB Cloud Training ofrece una solución sin código para entrenar modelos YOLO, perfecta para no programadores y dueños de negocios. El flujo de trabajo incluye la carga de conjuntos de datos, la selección de modelos y la configuración de instancias en la nube, lo que facilita el ajuste fino de los 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 notebook.
   ‍
2. Configuración y clave API: Configure la instalación e introduzca su clave API.
   ‍
3. Comenzar el entrenamiento: Ejecute los comandos de entrenamiento y su modelo comenzará a entrenarse.

Monitoreo del progreso del entrenamiento

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

Funciones e integraciones avanzadas

Entrenamiento en la nube e integración con RoboFlow

Ultralytics HUB ha introducido el entrenamiento en la nube, lo que permite a los usuarios entrenar modelos directamente en la nube sin necesidad de 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.

Aplicación móvil

La aplicación Ultralytics HUB lleva la detección de objetos en tiempo real a tu dispositivo móvil. Utilizando modelos pre-entrenados, la aplicación puede detectar objetos del conjunto de datos COCO con alta 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.

Resumiendo

Ultralytics HUB es un punto de inflexión en el mundo de la visión artificial, ya que facilita más que nunca el entrenamiento de modelos YOLO personalizados. Su interfaz intuitiva, sus robustas funciones y sus integraciones perfectas lo posicionan como una herramienta imprescindible tanto para desarrolladores como para investigadores. Tanto si busca entrenar modelos localmente, en la nube o sobre la marcha con la aplicación móvil, Ultralytics HUB le tiene cubierto.

Entonces, ¿por qué esperar? ¡Sumérjase en Ultralytics HUB hoy mismo y desbloquee el potencial de sus proyectos de aprendizaje automático con solo unos clics!