Yolo Vision Shenzhen
Shenzhen
Únete ahora

Segmentación de imágenes con Ultralytics YOLO11 en Google Colab

Abirami Vina

4 minutos de lectura

30 de diciembre de 2024

Descubra cómo utilizar eficazmente Ultralytics YOLO11 para la segmentación de imágenes, aprovechando un conjunto de datos de piezas de automóviles en Google Colab para realizar entrenamientos y pruebas sin interrupciones.

Modelos Ultralytics YOLO , como el último Ultralytics YOLO11admiten diversas tareas de visión artificial, como la detección de objetos, la clasificación de imágenes y la segmentación de instancias. Cada una de estas tareas pretende replicar un aspecto específico de la visión humana, haciendo posible que las máquinas vean e interpreten el mundo que las rodea. 

Por ejemplo, considere cómo un estudiante en una clase de arte puede coger un lápiz y delinear un objeto en un dibujo. Entre bastidores, su cerebro está realizando la segmentación, distinguiendo el objeto del fondo y otros elementos. La segmentación de imágenes logra un objetivo similar utilizando la inteligencia artificial (IA), descomponiendo los datos visuales en partes significativas para que las máquinas los entiendan. Esta técnica puede utilizarse en una variedad de aplicaciones en muchas industrias. 

__wf_reserved_inherit
Fig. 1. Ultralytics YOLO11 utilizado para segment objetos en una imagen.

Un ejemplo práctico es la segmentación de piezas de automóviles. Al identificar y clasificar los componentes específicos de un vehículo, la segmentación de imágenes puede agilizar los procesos en industrias como la fabricación de automóviles, la reparación y la catalogación de comercio electrónico.

En este artículo, exploraremos cómo puede utilizar Ultralytics YOLO11, Google Colab y el conjunto de datos Roboflow Carparts Segmentation para crear una solución que pueda identificar y segment con precisión las piezas de automóviles.

Ultralytics YOLO11 es fácil de usar

Ultralytics YOLO11 está disponible como modelo preentrenado a partir del conjunto de datosCOCO , que abarca 80 clases de objetos diferentes. Sin embargo, para aplicaciones específicas, como la segmentación de piezas de automóviles, el modelo puede entrenarse a medida para adaptarse mejor a su conjunto de datos y caso de uso. Esta flexibilidad permite a YOLO11 obtener buenos resultados tanto en tareas de uso general como en tareas muy especializadas.

El entrenamiento personalizado consiste en utilizar el modelo YOLO11 preentrenado y perfeccionarlo en un nuevo conjunto de datos. Al proporcionar ejemplos etiquetados específicos de su tarea, el modelo aprende a reconocer y segment objetos exclusivos de su proyecto. El entrenamiento personalizado garantiza una mayor precisión y relevancia en comparación con la utilización de pesos genéricos preentrenados.

La configuración de YOLO11 para el entrenamiento personalizado es sencilla. Con una configuración mínima, puede cargar el modelo y el conjunto de datos, iniciar el entrenamiento y supervisar métricas como la pérdida y la precisión durante el proceso. YOLO11 también incluye herramientas integradas para la validación y la evaluación, lo que facilita la evaluación del rendimiento del modelo. 

Ejecutar Ultralytics YOLO11 en Google Colab

Para la formación personalizada YOLO11, existen varias opciones para configurar un entorno. Una de las opciones más accesibles y cómodas es Google Colab. Estas son algunas de las ventajas de utilizar Google Colab para la formación de YOLO11 :

  • Acceso gratuito a los recursos: Google Colab proporciona GPU (unidades de procesamiento gráfico) y TPU (unidades de procesamientoTensor ), lo que permite entrenar YOLO11 sin necesidad de hardware costoso.
  • Entorno de colaboración: Google Colab te ayuda a compartir blocs de notas, a almacenar el trabajo en Google Drive y a simplificar el trabajo en equipo mediante una colaboración y un seguimiento de versiones sencillos.
  • Bibliotecas preinstaladas: Con herramientas preinstaladas como PyTorch y TensorFlow, Google Colab simplifica el proceso de configuración y te ayuda a empezar rápidamente.
  • Integración en la nube: Puedes cargar fácilmente conjuntos de datos desde Google Drive, GitHub u otras fuentes en la nube, lo que simplifica la preparación y el almacenamiento de datos.
__wf_reserved_inherit
Fig. 2. El cuaderno Google Colab YOLO11 .

Ultralytics también ofrece un cuadernoGoogle Colab preconfigurado específicamente para la formación de YOLO11 . Este cuaderno incluye todo lo necesario, desde la formación del modelo hasta la evaluación del rendimiento, lo que hace que el proceso sea sencillo y fácil de seguir. Es un buen punto de partida y le permite centrarse en ajustar el modelo a sus necesidades específicas sin preocuparse de complicados pasos de configuración.

Visión general del conjunto de datos de segmentación de piezas de automóvil Roboflow

Una vez decidido el entorno de formación, el siguiente paso es recopilar datos o elegir un conjunto de datos adecuado para segmentar piezas de automóviles. Elconjunto de datos RoboflowCarparts Segmentation Dataset, disponible en Roboflow Universe, es mantenido por Roboflow, una plataforma que proporciona herramientas para construir, entrenar y desplegar modelos de visión por ordenador. Este conjunto de datos incluye 3.156 imágenes de entrenamiento, 401 imágenes de validación y 276 imágenes de prueba, todas ellas con anotaciones de alta calidad de piezas de automóviles como parachoques, puertas, retrovisores y ruedas.

Normalmente, sería necesario descargar el conjunto de datos de Roboflow Universe y configurarlo manualmente para el entrenamiento en Google Collab. Sin embargo, el paqueteUltralytics Python simplifica este proceso al ofrecer una integración perfecta y herramientas preconfiguradas.

__wf_reserved_inherit
Fig 3. Ejemplos del conjunto de datos de segmentación de piezas de automóviles.

Con Ultralytics, el conjunto de datos está listo para su uso a través de un archivo YAML preconfigurado que incluye las rutas del conjunto de datos, las etiquetas de las clases y otros parámetros de entrenamiento. Esto se encarga de la configuración por usted, para que pueda cargar rápidamente el conjunto de datos y empezar directamente a entrenar su modelo. Además, el conjunto de datos está estructurado con conjuntos de entrenamiento, validación y prueba específicos, lo que facilita el seguimiento del progreso y la evaluación del rendimiento.

Al aprovechar el conjunto de datos de segmentación de Roboflow Carparts con las herramientas proporcionadas por Ultralytics YOLO11, dispondrá de un flujo de trabajo perfecto para crear modelos de segmentación de forma eficiente en plataformas como Google Colab. Este enfoque reduce el tiempo de configuración y le permite centrarse en perfeccionar su modelo para aplicaciones del mundo real.

Aplicaciones en el mundo real de la segmentación de piezas de automóviles

La segmentación de piezas de automóviles tiene una variedad de usos prácticos en diferentes industrias. Por ejemplo, en los talleres de reparación, puede ayudar a identificar y clasificar rápidamente los componentes dañados para que el proceso de reparación sea más rápido y eficiente. De manera similar, en la industria de seguros, los modelos de segmentación pueden automatizar las evaluaciones de reclamos analizando imágenes de vehículos dañados para identificar las piezas afectadas. Esto acelera el proceso de reclamos, reduce los errores y ahorra tiempo tanto a las aseguradoras como a los clientes.

__wf_reserved_inherit
Fig. 4. Segmentación de piezas de automóviles mediante YOLO.

Con respecto a la fabricación, la segmentación apoya el control de calidad al inspeccionar las piezas de automóviles en busca de defectos, garantizando la consistencia y reduciendo el desperdicio. Estas aplicaciones muestran cómo la segmentación de piezas de automóviles puede transformar las industrias al hacer que los procesos sean más seguros, rápidos y precisos.

Guía paso a paso: uso de YOLO11 en Google Colab 

Ahora que hemos cubierto todos los detalles, es el momento de ponerlo todo junto. Para empezar, puede consultar nuestro vídeo de YouTube, que le guiará por todo el proceso de configuración, formación y validación de un modelo YOLO11 para la segmentación de piezas de automóviles.

Aquí tienes una breve descripción general de los pasos involucrados:

  • Configura tu entorno en Google Colab: Activa la compatibilidad con GPU e instala el paquetePython Ultralytics para preparar el entrenamiento del modelo.
  • Cargue el modelo YOLO11 : Comience con un modelo de segmentación YOLO11 preentrenado para ahorrar tiempo y aprovechar las características existentes para la segmentación de piezas de automóviles.
  • Entrene el modelo con el conjunto de datos: Utilice el archivo "carparts-segyaml" durante el entrenamiento para descargar, configurar y utilizar automáticamente el conjunto de datos de segmentación de piezas de automóviles Roboflow . Ajuste parámetros como las épocas, el tamaño de la imagen y el tamaño del lote para afinar el modelo.
  • Supervise el progreso del entrenamiento: Realice un seguimiento de las principales métricas de rendimiento, como la pérdida de segmentación y la precisión mediamAP), para asegurarse de que el modelo mejora según lo esperado.
  • Validar y desplegar el modelo: Pruebe el modelo entrenado en el conjunto de validación para confirmar su precisión y expórtelo para aplicaciones del mundo real como el control de calidad o el procesamiento de reclamaciones de seguros.

Ventajas de utilizar YOLO11 para la segmentación de piezas de automóviles

YOLO11 es una herramienta fiable y eficaz para la segmentación de piezas de automóviles, que ofrece una serie de ventajas que la hacen ideal para diversas aplicaciones del mundo real. Estas son sus principales ventajas:

  • Velocidad y eficacia: YOLO11 procesa las imágenes con rapidez manteniendo una gran precisión, lo que la hace idónea para tareas en tiempo real como el control de calidad y los vehículos autónomos.
  • Alta precisión: El modelo destaca en la detección y segmentación de múltiples objetos dentro de una sola imagen, lo que garantiza una identificación precisa de las piezas del automóvil.
  • Escalabilidad: YOLO11 puede manejar grandes conjuntos de datos y complejas tareas de segmentación, lo que lo hace escalable para aplicaciones industriales.
  • Múltiples integraciones: Ultralytics admite integraciones con plataformas como Google Colab, Ultralytics Hub y otras herramientas populares, lo que mejora la flexibilidad y la accesibilidad para los desarrolladores.

Consejos para trabajar con YOLO11 en Google Collab

Aunque Google Colab hace que los flujos de trabajo de aprendizaje automático sean mucho más fáciles de manejar, puede llevar un poco de tiempo acostumbrarse a él si eres nuevo. Navegar por la configuración basada en la nube, los ajustes de tiempo de ejecución y los límites de sesión puede resultar complicado al principio, pero hay algunos consejos que pueden facilitar mucho las cosas.

Aquí hay algunas consideraciones que debe tener en cuenta:

  • Empieza por activar la aceleración de GPU en los ajustes de ejecución para acelerar el entrenamiento.
  • Dado que Colab se ejecuta en la nube, asegúrate de tener una conexión a Internet estable para acceder a recursos como conjuntos de datos y repositorios. 
  • Organiza tus archivos y conjuntos de datos en Google Drive o GitHub para facilitar su carga y gestión en Colab.
  • Si tiene limitaciones de memoria en el nivel gratuito de Colab, intente reducir el tamaño de la imagen o el tamaño del lote durante el entrenamiento.
  • Recuerda guardar tu modelo y resultados regularmente, ya que las sesiones de Colab tienen límites de tiempo y no querrás perder tu progreso. 

Consigue más con YOLO11

Ultralytics YOLO11, combinado con plataformas como Google Colab y conjuntos de datos como el Roboflow Carparts Segmentation dataset, hace que la segmentación de imágenes sea sencilla y accesible. Con sus herramientas intuitivas, sus modelos preentrenados y su sencilla configuración, YOLO11 le permite sumergirse en tareas avanzadas de visión por ordenador con facilidad. 

Tanto si está mejorando la seguridad en automoción, optimizando la fabricación o creando aplicaciones de IA innovadoras, esta combinación le proporciona las herramientas que le ayudarán a tener éxito. Con Ultralytics YOLO11, no solo construirá modelos, sino que allanará el camino hacia soluciones más inteligentes y eficientes en el mundo real.

Para obtener más información, consulta nuestro repositorio de GitHub y participa en nuestra comunidad. Explora las aplicaciones de la IA en coches autónomos y la visión artificial para la agricultura en nuestras páginas de soluciones. 🚀

¡Construyamos juntos el futuro
de la IA!

Comience su viaje con el futuro del aprendizaje automático

Comienza gratis