Validación de modelos con Ultralytics YOLO

Introducción

La validación es un paso crítico en el pipeline de aprendizaje automático, que le permite evaluar la calidad de sus modelos entrenados. El modo Val en Ultralytics YOLO11 proporciona un conjunto robusto de herramientas y métricas para evaluar el rendimiento de sus modelos de detección de objetos. Esta guía sirve como un recurso completo para entender cómo usar eficazmente el modo Val para asegurar que sus modelos sean precisos y fiables.

Ver: Tutorial de modos de Ultralytics: Validación

¿Por qué validar con Ultralytics YOLO?

He aquí por qué es ventajoso usar el modo Val de YOLO11:

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

Características clave del modo Val

Estas son las funcionalidades notables que ofrece el modo Val de YOLO11:

• Configuración automatizada: Los modelos recuerdan sus configuraciones de entrenamiento para una validación sencilla.
• Soporte multi-métrico: Evalúe su modelo basándose en una variedad de métricas de precisión.
• CLI y API de python: Elija entre la interfaz de línea de comandos o la API de python según su preferencia para la 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

Validar el modelo YOLO11n entrenado precisión en el conjunto de datos COCO8. No se necesitan argumentos ya que el model conserva su entrenamiento data y argumentos como atributos del modelo. Consulte la sección Argumentos a continuación para obtener una lista completa de los argumentos de validación.

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n.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=yolo11n.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 presenta un desglose detallado de cada argumento para ayudarle a personalizar su configuración de validación de forma eficaz.

Cada una de estas configuraciones juega un papel vital en el proceso de validación, permitiendo una evaluación personalizable y eficiente de los modelos YOLO. Ajustar estos parámetros de acuerdo con sus necesidades y recursos específicos puede ayudar a lograr el mejor equilibrio entre precisión y rendimiento.

Ejemplo de validación con argumentos

Ver: Cómo exportar resultados de validación de modelos en CSV, JSON, SQL, Polars DataFrame y más

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("yolo11n.pt")

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

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

from ultralytics import YOLO

model = YOLO("yolo11n.pt")

results = model.val(data="coco8.yaml", plots=True)
print(results.confusion_matrix.to_df())

Para obtener más detalles, consulte la DataExportMixin documentación de la clase.

Preguntas frecuentes

¿Cómo valido mi modelo YOLO11 con Ultralytics?

Para validar su modelo YOLO11, puede utilizar el modo Val proporcionado por Ultralytics. Por ejemplo, utilizando la API de python, puede cargar un modelo y ejecutar la validación con:

from ultralytics import YOLO

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

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

Alternativamente, puede utilizar la interfaz de línea de comandos (CLI):

yolo val model=yolo11n.pt

Para una mayor personalización, puede ajustar varios argumentos como imgsz, batch, y conf tanto en los modos python como en CLI. Consulte la Argumentos para la validación del modelo YOLO sección para obtener la lista completa de parámetros.

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

La validación del modelo YOLO11 proporciona varias métricas clave para evaluar el rendimiento del modelo. Éstas incluyen:

• mAP50 (Precisión Media Promedio en el umbral IoU 0.5)
• mAP75 (Precisión Media Promedio en el umbral IoU 0.75)
• mAP50-95 (Precisión Media Promedio en múltiples umbrales IoU de 0.5 a 0.95)

Usando la API de python, puede acceder a estas métricas de la siguiente manera:

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, consulte las Características clave del modo Val.

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

El uso de Ultralytics YOLO para la validación proporciona varias ventajas:

• Precisión: YOLO11 ofrece métricas de rendimiento precisas, incluyendo mAP50, mAP75 y mAP50-95.
• Comodidad: Los modelos recuerdan su configuración de entrenamiento, lo que facilita la validación.
• Flexibilidad: Puede realizar la validación con los mismos conjuntos de datos y tamaños de imagen o con otros diferentes.
• Ajuste de hiperparámetros: Las métricas de validación ayudan a ajustar los modelos para obtener un mejor rendimiento.

Estos beneficios aseguran que sus modelos se evalúen a fondo y puedan optimizarse para obtener resultados superiores. Obtenga más información sobre estas ventajas en la sección Por qué validar con Ultralytics YOLO.

¿Puedo validar mi modelo YOLO11 usando un conjunto de datos personalizado?

Sí, puede validar su modelo YOLO11 utilizando un conjunto de datos personalizado. Especifique el data argumento con la ruta a su archivo de configuración del conjunto de datos. Este archivo debe incluir rutas a los datos de validación, nombres de clase y otros detalles relevantes.

Ejemplo en Python:

from ultralytics import YOLO

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

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

Ejemplo usando CLI:

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

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

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

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

Ejemplo en Python:

from ultralytics import YOLO

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

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

Ejemplo usando CLI:

yolo val model=yolo11n.pt save_json=True

Esta funcionalidad es particularmente útil para un análisis más profundo o la integración con otras herramientas. Consulte los Argumentos para la validación del modelo YOLO para obtener más detalles.

📅 Creado hace 1 año ✏️ Actualizado hace 5 días

Comentarios