Elevando YOLOv8 Formación: Simplifica tu proceso de registro con Comet ML

Registrar los detalles clave del entrenamiento, como parámetros, métricas, predicciones de imágenes y puntos de comprobación del modelo, es esencial en el aprendizaje automático: mantiene la transparencia de tu proyecto, tu progreso medible y tus resultados repetibles.

Ultralytics YOLOv8 se integra perfectamente con Comet ML, capturando y optimizando eficazmente cada aspecto del proceso de entrenamiento de tu modelo de detección de objetos YOLOv8 . En esta guía, cubriremos el proceso de instalación, la configuración de Comet ML, la información en tiempo real, el registro personalizado y el uso sin conexión, garantizando que tu entrenamiento de YOLOv8 esté completamente documentado y ajustado para obtener resultados sobresalientes.

Comet ML

Comet ML es una plataforma para seguir, comparar, explicar y optimizar modelos y experimentos de aprendizaje automático. Te permite registrar métricas, parámetros, medios y mucho más durante el entrenamiento de tus modelos y supervisar tus experimentos a través de una interfaz web estéticamente agradable. Comet ML ayuda a los científicos de datos a iterar más rápidamente, mejora la transparencia y la reproducibilidad, y ayuda en el desarrollo de modelos de producción.

Aprovechar el poder de YOLOv8 y Comet ML

Al combinar Ultralytics YOLOv8 con Comet ML, desbloqueas una serie de ventajas. Entre ellas se incluyen una gestión simplificada de los experimentos, información en tiempo real para realizar ajustes rápidos, opciones de registro flexibles y personalizadas, y la posibilidad de registrar experimentos sin conexión cuando el acceso a Internet es limitado. Esta integración te permite tomar decisiones basadas en datos, analizar las métricas de rendimiento y lograr resultados excepcionales.

Instalación

Para instalar los paquetes necesarios, ejecuta

# Install the required packages for YOLOv8 and Comet ML
pip install ultralytics comet_ml torch torchvision

Configuración de Comet ML

Después de instalar los paquetes necesarios, tendrás que registrarte, obtener una Clave APIComet y configurarla.

# Set your Comet Api Key
export COMET_API_KEY=<Your API Key>

A continuación, puedes inicializar tu proyecto Comet . Comet detectará automáticamente la clave API y procederá a la configuración.

import comet_ml

comet_ml.init(project_name="comet-example-yolov8-coco128")

Si utilizas un cuaderno Colab Google , el código anterior te pedirá que introduzcas tu clave API para la inicialización.

Utilización

Antes de sumergirte en las instrucciones de uso, asegúrate de consultar la gama de modelosYOLOv8 que ofrece Ultralytics. Esto te ayudará a elegir el modelo más apropiado para los requisitos de tu proyecto.

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n.pt")

# train the model
results = model.train(
    data="coco8.yaml",
    project="comet-example-yolov8-coco128",
    batch=32,
    save_period=1,
    save_json=True,
    epochs=3,
)

Tras ejecutar el código de entrenamiento, Comet ML creará un experimento en tu espacio de trabajo Comet para realizar un seguimiento automático de la ejecución. A continuación, se te proporcionará un enlace para ver el registro detallado del proceso de entrenamiento de tu modeloYOLOv8 .

Comet registra automáticamente los siguientes datos sin ninguna configuración adicional: métricas como mAP y pérdida, hiperparámetros, puntos de control del modelo, matriz de confusión interactiva y predicciones del cuadro delimitador de la imagen.

Comprende el rendimiento de tu modelo con Comet Visualizaciones ML

Profundicemos en lo que verás en el Comet Panel de ML una vez que su YOLOv8 El modelo comienza a entrenarse. El tablero es donde ocurre toda la acción, presentando una variedad de información registrada automáticamente a través de imágenes y estadísticas. Aquí hay un recorrido rápido:

Paneles de experimentación

La sección de paneles de experimentos del panel Comet ML organiza y presenta las distintas ejecuciones y sus métricas, como la pérdida de máscara de segmento, la pérdida de clase, la precisión y la precisión media promedio.

Métricas

En la sección de métricas, tienes la opción de examinar las métricas también en formato tabular, que se muestra en un panel específico, como se ilustra aquí.

Matriz de confusión interactiva

La matriz de confusión, que se encuentra en la pestaña Matriz de confusión, proporciona una forma interactiva de evaluar la precisión de clasificación del modelo. Detalla las predicciones correctas e incorrectas, permitiéndote comprender los puntos fuertes y débiles del modelo.

Métricas del sistema

Comet ML registra métricas del sistema para ayudar a identificar cualquier cuello de botella en el proceso de entrenamiento. Incluye métricas como la utilización de GPU , el uso de memoria de GPU , la utilización de CPU y el uso de RAM. Son esenciales para controlar la eficiencia del uso de recursos durante el entrenamiento del modelo.

Personaliza el registro de Comet ML

Comet ML ofrece la flexibilidad de personalizar su comportamiento de registro estableciendo variables de entorno. Estas configuraciones te permiten adaptar Comet ML a tus necesidades y preferencias específicas. Aquí tienes algunas opciones de personalización útiles:

Registro de predicciones de imagen

Puedes controlar el número de predicciones de imágenes que Comet ML registra durante tus experimentos. Por defecto, Comet ML registra 100 predicciones de imágenes del conjunto de validación. Sin embargo, puedes cambiar este número para adaptarlo mejor a tus necesidades. Por ejemplo, para registrar 200 predicciones de imagen, utiliza el siguiente código:

import os

os.environ["COMET_MAX_IMAGE_PREDICTIONS"] = "200"

Intervalo de registro por lotes

Comet ML te permite especificar la frecuencia con la que se registran los lotes de predicciones de imágenes. La dirección COMET_EVAL_BATCH_LOGGING_INTERVAL controla esta frecuencia. El valor por defecto es 1, que registra las predicciones de cada lote de validación. Puedes ajustar este valor para registrar las predicciones en un intervalo diferente. Por ejemplo, si lo ajustas a 4, se registrarán las predicciones de cada cuarto lote.

import os

os.environ["COMET_EVAL_BATCH_LOGGING_INTERVAL"] = "4"

Desactivar el registro de la matriz de confusión

En algunos casos, puede que no quieras registrar la matriz de confusión de tu conjunto de validación después de cada época. Puedes desactivar esta función configurando la opción COMET_EVAL_LOG_CONFUSION_MATRIX a "falso". La matriz de confusión sólo se registrará una vez, una vez finalizado el entrenamiento.

import os

os.environ["COMET_EVAL_LOG_CONFUSION_MATRIX"] = "false"

Registro sin conexión

Si te encuentras en una situación en la que el acceso a Internet es limitado, Comet ML proporciona una opción de registro sin conexión. Puedes configurar el COMET_MODE a "sin conexión" para activar esta función. Los datos de tu experimento se guardarán localmente en un directorio que podrás cargar más tarde en Comet ML cuando dispongas de conexión a Internet.

import os

os.environ["COMET_MODE"] = "offline"

Resumen

Esta guía te ha guiado a través de la integración de Comet ML con Ultralytics' YOLOv8. Desde la instalación hasta la personalización, has aprendido a agilizar la gestión de experimentos, obtener información en tiempo real y adaptar el registro a las necesidades de tu proyecto.

Explora la documentación oficial deComet ML para obtener más información sobre la integración con YOLOv8.

Además, si quieres profundizar en las aplicaciones prácticas de YOLOv8, en concreto para tareas de segmentación de imágenes, esta guía detallada sobre el ajuste fino de YOLOv8 con Comet ML ofrece valiosos conocimientos e instrucciones paso a paso para mejorar el rendimiento de tu modelo.

Además, para explorar otras interesantes integraciones con Ultralytics, consulta la página de la guía de integración, que ofrece una gran cantidad de recursos e información.

PREGUNTAS FRECUENTES

¿Cómo integro Comet ML con Ultralytics YOLOv8 para el entrenamiento?

Para integrar Comet ML con Ultralytics YOLOv8 , sigue estos pasos:

1. Instala los paquetes necesarios:

   pip install ultralytics comet_ml torch torchvision
   

2. Configura tu clave API en Comet :

   export COMET_API_KEY=<Your API Key>
   

3. Inicializa tu proyecto Comet en tu código Python :

   import comet_ml
   
   comet_ml.init(project_name="comet-example-yolov8-coco128")
   

4. Entrena tu modelo YOLOv8 y registra las métricas:

   from ultralytics import YOLO
   
   model = YOLO("yolov8n.pt")
   results = model.train(
       data="coco8.yaml", project="comet-example-yolov8-coco128", batch=32, save_period=1, save_json=True, epochs=3
   )
   

Para obtener instrucciones más detalladas, consulta la sección de configuración deComet ML.

¿Cuáles son las ventajas de utilizar Comet ML con YOLOv8?

Integrando Ultralytics YOLOv8 con Comet ML, puedes:

• Controla la información en tiempo real: Obtén información instantánea sobre los resultados de tu entrenamiento, lo que permite realizar ajustes rápidos.
• Registra métricas exhaustivas: Captura automáticamente métricas esenciales como mAP, pérdida, hiperparámetros y puntos de control del modelo.
• Realiza el seguimiento de los experimentos sin conexión: Registra localmente tus recorridos de entrenamiento cuando no dispongas de acceso a Internet.
• Compara diferentes ejecuciones de entrenamiento: Utiliza el panel interactivo Comet ML para analizar y comparar múltiples experimentos.

Aprovechando estas funciones, puedes optimizar tus flujos de trabajo de aprendizaje automático para mejorar el rendimiento y la reproducibilidad. Para más información, visita la guía de integración deComet ML.

¿Cómo personalizo el comportamiento de registro de Comet ML durante el entrenamiento de YOLOv8 ?

Comet ML permite una amplia personalización de su comportamiento de registro mediante variables de entorno:

• Cambia el número de predicciones de imagen registradas:

  import os
  
  os.environ["COMET_MAX_IMAGE_PREDICTIONS"] = "200"
  

• Ajusta el intervalo de registro por lotes:

  import os
  
  os.environ["COMET_EVAL_BATCH_LOGGING_INTERVAL"] = "4"
  

• Desactiva el registro de la matriz de confusión:

  import os
  
  os.environ["COMET_EVAL_LOG_CONFUSION_MATRIX"] = "false"
  

Para más opciones de personalización, consulta la sección Personalizar el registro de Comet ML.

¿Cómo puedo ver métricas y visualizaciones detalladas de mi entrenamiento en YOLOv8 en Comet ML?

Una vez que tu modelo YOLOv8 empiece a entrenarse, podrás acceder a una amplia gama de métricas y visualizaciones en el panel Comet ML. Entre las principales funciones se incluyen:

• Paneles de experimentos: Visualiza diferentes ejecuciones y sus métricas, incluyendo la pérdida de máscara de segmento, la pérdida de clase y la precisión media promedio.
• Métricas: Examina las métricas en formato tabular para un análisis detallado.
• Matriz de Confusión Interactiva: Evalúa la precisión de la clasificación con una matriz de confusión interactiva.
• Métricas del sistema: Supervisa la utilización de GPU y CPU , el uso de memoria y otras métricas del sistema.

Para una visión detallada de estas funciones, visita la sección Comprender el rendimiento de tu modelo con las visualizaciones de Comet ML.

¿Puedo utilizar Comet ML para el registro sin conexión al entrenar modelos YOLOv8 ?

Sí, puedes activar el registro sin conexión en Comet ML configurando COMET_MODE a "desconectado":

import os

os.environ["COMET_MODE"] = "offline"

Esta función te permite registrar localmente los datos de tus experimentos, que más tarde podrás cargar en Comet ML cuando dispongas de conexión a Internet. Esto es especialmente útil cuando se trabaja en entornos con acceso limitado a Internet. Para más detalles, consulta la sección Registro sin conexión.

Comentarios