Validación del modelo con Ultralytics YOLO

Introducción

La validación es un paso crítico en el proceso de aprendizaje automático, que te permite evaluar la calidad de tus modelos entrenados. El modo Val de Ultralytics YOLOv8 proporciona un sólido conjunto de herramientas y métricas para evaluar el rendimiento de tus modelos de detección de objetos. Esta guía es un recurso completo para comprender cómo utilizar eficazmente el modo Val para asegurarte de que tus modelos son precisos y fiables.

Observa: Ultralytics Tutorial de Modos: Validación

¿Por qué validar con Ultralytics YOLO ?

He aquí por qué es ventajoso utilizar el modo Val de YOLOv8:

• Precisión: Obtén métricas precisas como mAP50, mAP75 y mAP50-95 para evaluar exhaustivamente tu modelo.
• Comodidad: Utiliza funciones integradas que recuerden los ajustes de entrenamiento, simplificando el proceso de validación.
• Flexibilidad: Valida tu modelo con los mismos o diferentes conjuntos de datos y tamaños de imagen.
• Ajuste de hiperparámetros: Utiliza las métricas de validación para ajustar tu modelo y obtener un mejor rendimiento.

Características principales del modo Val

Estas son las funcionalidades destacadas que ofrece el modo Val de YOLOv8:

• Configuraciones automatizadas: Los modelos recuerdan sus configuraciones de entrenamiento para una validación sencilla.
• Soporte multimétrico: Evalúa tu modelo basándote en una serie de métricas de precisión.
• CLI y la API Python : Elige entre la interfaz de línea de comandos o la API Python en función de tus preferencias de validación.
• Compatibilidad de datos: Funciona perfectamente con los conjuntos de datos utilizados durante la fase de entrenamiento, así como con conjuntos de datos personalizados.

Ejemplos de uso

Valida la precisión del modelo entrenado YOLOv8n en el conjunto de datos COCO8. No es necesario pasar ningún argumento como model conserva su formación data y los argumentos como atributos del modelo. Consulta la sección Argumentos para ver una lista completa de los argumentos de exportación.

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom model

# Validate the model
metrics = model.val()  # no arguments needed, dataset and settings remembered
metrics.box.map  # map50-95
metrics.box.map50  # map50
metrics.box.map75  # map75
metrics.box.maps  # a list contains map50-95 of each category

yolo detect val model=yolov8n.pt  # val official model
yolo detect val model=path/to/best.pt  # val custom model

Argumentos para la validación del modelo YOLO

Al validar los modelos YOLO , se pueden ajustar varios argumentos para optimizar el proceso de evaluación. Estos argumentos controlan aspectos como el tamaño de la imagen de entrada, el procesamiento por lotes y los umbrales de rendimiento. A continuación se ofrece un desglose detallado de cada argumento para ayudarte a personalizar eficazmente tus ajustes de validación.

Cada uno de estos ajustes desempeña un papel vital en el proceso de validación, permitiendo una evaluación personalizable y eficaz de los modelos YOLO . Ajustar estos parámetros según tus necesidades y recursos específicos puede ayudarte a conseguir el mejor equilibrio entre precisión y rendimiento.

Ejemplo de validación con argumentos

Los siguientes ejemplos muestran la validación del modelo YOLO con argumentos personalizados en Python y CLI.

from ultralytics import YOLO

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

# Customize validation settings
validation_results = model.val(data="coco8.yaml", imgsz=640, batch=16, conf=0.25, iou=0.6, device="0")

yolo val model=yolov8n.pt data=coco8.yaml imgsz=640 batch=16 conf=0.25 iou=0.6 device=0

PREGUNTAS FRECUENTES

¿Cómo puedo validar mi modelo YOLOv8 con Ultralytics?

Para validar tu modelo YOLOv8 , puedes utilizar el modo Val proporcionado por Ultralytics. Por ejemplo, utilizando la API Python , puedes cargar un modelo y ejecutar la validación con:

from ultralytics import YOLO

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

# Validate the model
metrics = model.val()
print(metrics.box.map)  # map50-95

También puedes utilizar la interfaz de línea de comandos (CLI):

yolo val model=yolov8n.pt

Para una mayor personalización, puedes ajustar varios argumentos como imgsz, batchy conf en los modos Python y CLI . Comprueba Argumentos para la validación del modelo YOLO para ver la lista completa de parámetros.

¿Qué métricas puedo obtener de la validación del modelo YOLOv8 ?

YOLOv8 La validación del modelo proporciona varias métricas clave para evaluar el rendimiento del modelo. Entre ellas se incluyen:

• mAP50 (Precisión media en el umbral IoU 0,5)
• mAP75 (Precisión media media en el umbral IoU 0,75)
• mAP50-95 (Precisión media a través de múltiples umbrales de IoU de 0,5 a 0,95)

Utilizando la API Python , puedes acceder a estas métricas del siguiente modo:

metrics = model.val()  # assumes `model` has been loaded
print(metrics.box.map)  # mAP50-95
print(metrics.box.map50)  # mAP50
print(metrics.box.map75)  # mAP75
print(metrics.box.maps)  # list of mAP50-95 for each category

Para una evaluación completa del rendimiento, es crucial revisar todas estas métricas. Para más detalles, consulta las Características principales del Modo Val.

¿Cuáles son las ventajas de utilizar Ultralytics YOLO para la validación?

Utilizar Ultralytics YOLO para la validación ofrece varias ventajas:

• Precisión: YOLOv8 ofrece métricas de rendimiento precisas, como mAP50, mAP75 y mAP50-95.
• Comodidad: Los modelos recuerdan sus ajustes de entrenamiento, lo que facilita la validación.
• Flexibilidad: Puedes validar con los mismos o diferentes conjuntos de datos y tamaños de imagen.
• Ajuste de hiperparámetros: Las métricas de validación ayudan a ajustar los modelos para mejorar su rendimiento.

Estas ventajas garantizan que tus modelos se evalúen a fondo y puedan optimizarse para obtener resultados superiores. Obtén más información sobre estas ventajas en la sección Por qué validar con Ultralytics YOLO .

¿Puedo validar mi modelo YOLOv8 utilizando un conjunto de datos personalizado?

Sí, puedes validar tu modelo YOLOv8 utilizando un conjunto de datos personalizado. Especifica el data con la ruta a tu archivo de configuración del conjunto de datos. Este archivo debe incluir las rutas a los datos de validación, los nombres de las clases y otros detalles relevantes.

Ejemplo en Python:

from ultralytics import YOLO

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

# Validate with a custom dataset
metrics = model.val(data="path/to/your/custom_dataset.yaml")
print(metrics.box.map)  # map50-95

Ejemplo utilizando CLI:

yolo val model=yolov8n.pt data=path/to/your/custom_dataset.yaml

Para más opciones personalizables durante la validación, consulta la sección Ejemplo de validación con argumentos.

¿Cómo guardo los resultados de la validación en un archivo JSON en YOLOv8?

Para guardar los resultados de la validación en un archivo JSON, puedes establecer la opción save_json argumento para True al ejecutar la validación. Esto puede hacerse tanto en la API Python como en CLI.

Ejemplo en Python:

from ultralytics import YOLO

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

# Save validation results to JSON
metrics = model.val(save_json=True)

Ejemplo utilizando CLI:

yolo val model=yolov8n.pt save_json=True

Esta funcionalidad es especialmente útil para análisis posteriores o para la integración con otras herramientas. Consulta los Argumentos para la validación del modelo YOLO para más detalles.

Comentarios