Modelos admitidos por Ultralytics
Bienvenido a la documentación sobre modelos de Ultralytics Ofrecemos soporte para una amplia gama de modelos, cada uno adaptado a tareas específicas como la detección de objetos, la segmentación de instancias, la clasificación de imágenes, la estimación de poses y el seguimiento multiobjeto. Si estás interesado en contribuir con la arquitectura de tu modelo a Ultralytics, consulta nuestra Guía de contribución.
Modelos destacados
Estos son algunos de los principales modelos compatibles:
- YOLOv3: La tercera iteración de la familia de modelos YOLO , original de Joseph Redmon, conocida por su eficaz capacidad de detección de objetos en tiempo real.
- YOLOv4: Una actualización nativa de la red oscura de YOLOv3, publicada por Alexey Bochkovskiy en 2020.
- YOLOv5: Una versión mejorada de la arquitectura YOLO de Ultralytics, que ofrece mejores prestaciones y velocidad en comparación con las versiones anteriores.
- YOLOv6: Lanzado por Meituan en 2022, y en uso en muchos de los robots autónomos de reparto de la empresa.
- YOLOv7: Modelos actualizados de YOLO publicados en 2022 por los autores de YOLOv4.
- YOLOv8: La última versión de la familia YOLO , con funciones mejoradas como la segmentación de instancias, la estimación de poses/puntos clave y la clasificación.
- YOLOv9: Un modelo experimental entrenado en el Ultralytics YOLOv5 código base que implementa la Información de Gradiente Programable (PGI).
- YOLOv10: Por la Universidad de Tsinghua, con un entrenamiento sin NMS y una arquitectura orientada a la eficiencia y la precisión, que ofrece un rendimiento y una latencia de vanguardia.
- YOLO11 🚀 NUEVO: Ultralytics' últimos modelos YOLO que ofrecen un rendimiento de vanguardia (SOTA) en múltiples tareas.
- Modelo de segmento de cualquier cosa (SAM): El modelo original de Segment Anything de Meta (SAM).
- Modelo 2 de cualquier segmento (SAM2): La nueva generación del Segment Anything Model de Meta (SAM) para vídeos e imágenes.
- Modelo de cualquier cosa para segmentos móviles (MobileSAM): MobileSAM para aplicaciones móviles, por la Universidad Kyung Hee.
- Modelo Rápido de Cualquier Segmento (FastSAM): FastSAM por el Grupo de Análisis de Imagen y Vídeo del Instituto de Automatización de la Academia China de las Ciencias.
- YOLO-NAS: YOLO Modelos de búsqueda de arquitectura neuronal (NAS).
- Transformadores de Detección en Tiempo Real (RT-DETR): Modelos de transformadores de detección en tiempo real de Baidu PaddlePaddle (RT-DETR).
- YOLO-Mundo: Modelos de Detección de Objetos de Vocabulario Abierto en tiempo real del Laboratorio de IA de Tencent.
Observa: Ejecuta los modelos de Ultralytics YOLO con sólo unas líneas de código.
Primeros pasos: Ejemplos de uso
Este ejemplo proporciona ejemplos sencillos de entrenamiento e inferencia de YOLO . Para obtener documentación completa sobre estos y otros modos, consulta las páginas de documentación Predecir, Entrenar, Val y Exportar.
Ten en cuenta que el ejemplo siguiente es para los modelos YOLOv8 Detectar para la detección de objetos. Para otras tareas compatibles, consulta los documentos Segmentar, Clasificar y Pose.
Ejemplo
PyTorch preentrenado *.pt
modelos, así como la configuración *.yaml
pueden pasarse a la función YOLO()
, SAM()
, NAS()
y RTDETR()
para crear una instancia del modelo en Python:
from ultralytics import YOLO
# Load a COCO-pretrained YOLOv8n model
model = YOLO("yolov8n.pt")
# Display model information (optional)
model.info()
# Train the model on the COCO8 example dataset for 100 epochs
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)
# Run inference with the YOLOv8n model on the 'bus.jpg' image
results = model("path/to/bus.jpg")
CLI para ejecutar directamente los modelos:
Contribuir con nuevos modelos
¿Interesado en contribuir con tu modelo a Ultralytics? ¡Estupendo! Siempre estamos abiertos a ampliar nuestra cartera de modelos.
Bifurca el repositorio: Comienza bifurcando el repositorio GitHub deUltralytics .
Clona tu bifurcación: Clona tu bifurcación en tu máquina local y crea una nueva rama en la que trabajar.
Implementa tu modelo: Añade tu modelo siguiendo las normas y directrices de codificación proporcionadas en nuestra Guía de Contribución.
Prueba a fondo: Asegúrate de probar tu modelo rigurosamente, tanto de forma aislada como formando parte del pipeline.
Crea una Pull Request: Una vez que estés satisfecho con tu modelo, crea una pull request al repositorio principal para su revisión.
Revisión del código y fusión: Tras la revisión, si tu modelo cumple nuestros criterios, se fusionará en el repositorio principal.
Para conocer los pasos detallados, consulta nuestra Guía de Contribución.
PREGUNTAS FRECUENTES
¿Cuáles son las principales ventajas de utilizar Ultralytics YOLOv8 para la detección de objetos?
Ultralytics YOLOv8 ofrece funciones mejoradas como la detección de objetos en tiempo real, la segmentación de instancias, la estimación de poses y la clasificación. Su arquitectura optimizada garantiza un rendimiento de alta velocidad sin sacrificar la precisión, por lo que es ideal para una gran variedad de aplicaciones. YOLOv8 también incluye compatibilidad integrada con conjuntos de datos y modelos populares, como se detalla en la página de documentaciónYOLOv8 .
¿Cómo puedo entrenar un modelo YOLOv8 con datos personalizados?
El entrenamiento de un modelo YOLOv8 con datos personalizados puede realizarse fácilmente utilizando las bibliotecas Ultralytics'. He aquí un ejemplo rápido:
Ejemplo
Para obtener instrucciones más detalladas, visita la página de documentación del Tren.
¿Qué versiones de YOLO son compatibles con Ultralytics?
Ultralytics es compatible con una amplia gama de versiones de YOLO (You Only Look Once), desde YOLOv3 hasta YOLOv10, junto con modelos como NAS, SAM y RT-DETR. Cada versión está optimizada para diversas tareas, como la detección, la segmentación y la clasificación. Para obtener información detallada sobre cada modelo, consulta la documentación Modelos admitidos por Ultralytics.
¿Por qué debería utilizar Ultralytics HUB para proyectos de aprendizaje automático?
Ultralytics HUB proporciona una plataforma integral sin código para formar, desplegar y gestionar los modelos YOLO . Simplifica los flujos de trabajo complejos, permitiendo a los usuarios centrarse en el rendimiento y la aplicación de los modelos. El HUB también ofrece capacidades de entrenamiento en la nube, gestión integral de conjuntos de datos e interfaces fáciles de usar. Obtén más información en la página de documentación de Ultralytics HUB.
¿Qué tipos de tareas puede realizar YOLOv8 , y cómo se compara con otras versiones de YOLO ?
YOLOv8 es un modelo versátil capaz de realizar tareas como la detección de objetos, la segmentación de instancias, la clasificación y la estimación de poses. En comparación con versiones anteriores como YOLOv3 y YOLOv4, YOLOv8 ofrece mejoras significativas en velocidad y precisión gracias a su arquitectura optimizada. Para una comparación más profunda, consulta la documentación deYOLOv8 y las páginas de Tareas para obtener más detalles sobre tareas específicas.