CCW-YOLO: Una red YOLOv5s modificada para la detección de peatones en escenas de tráfico complejas
Autores: Wang, Zhaodi; Yang, Shuqiang; Qin, Huafeng; Liu, Yike; Ding, Jinyan
Idioma: Inglés
Editor: MDPI
Año: 2024
Acceso abierto
Artículo científico
2024
CCW-YOLO: Una red YOLOv5s modificada para la detección de peatones en escenas de tráfico complejas
Categoría
Gestión y administración
Subcategoría
Gestión de la tecnología y la inovación
Palabras clave
Escenas de tráfico
Detección de objetivos peatonales
Algoritmo CCW-YOLO
GhostConv
Módulo C2f
Módulo de Atención por Coordenadas
Licencia
CC BY-SA – Atribución – Compartir Igual
Consultas: 1
Citaciones: Sin citaciones
En las escenas de tráfico, la detección de objetivos peatonales enfrenta problemas significativos de detección errónea y omisión debido a factores como la densidad de la multitud y la oclusión de obstáculos. Para abordar estos desafíos y mejorar la precisión de la detección, proponemos un algoritmo mejorado CCW-YOLO. El algoritmo primero introduce una capa convolucional ligera utilizando GhostConv e incorpora un módulo C2f mejorado para mejorar el rendimiento de detección de la red. Además, integra el módulo de Atención de Coordenadas para capturar mejor los puntos clave de los objetivos. A continuación, la función de pérdida de la caja delimitadora CIoU en la salida de YOLOv5 se reemplaza con la pérdida WiseIoU para mejorar la adaptabilidad a varios escenarios de detección, mejorando así aún más la precisión. Finalmente, desarrollamos un sistema de detección de conteo de peatones utilizando PyQt5 para mejorar la interacción humano-computadora. Los resultados experimentales en el conjunto de datos público INRIA mostraron que nuestro algoritmo logró una precisión de detección del 98.4%, lo que representa una mejora del 10.1% sobre el algoritmo original YOLOv5s. Este avance mejora significativamente la detección de objetos pequeños en imágenes y aborda de manera efectiva los problemas de detección errónea y omisión en entornos complejos. Estos hallazgos tienen importantes implicaciones prácticas para garantizar la seguridad vial y optimizar el flujo de tráfico.
Descripción
En las escenas de tráfico, la detección de objetivos peatonales enfrenta problemas significativos de detección errónea y omisión debido a factores como la densidad de la multitud y la oclusión de obstáculos. Para abordar estos desafíos y mejorar la precisión de la detección, proponemos un algoritmo mejorado CCW-YOLO. El algoritmo primero introduce una capa convolucional ligera utilizando GhostConv e incorpora un módulo C2f mejorado para mejorar el rendimiento de detección de la red. Además, integra el módulo de Atención de Coordenadas para capturar mejor los puntos clave de los objetivos. A continuación, la función de pérdida de la caja delimitadora CIoU en la salida de YOLOv5 se reemplaza con la pérdida WiseIoU para mejorar la adaptabilidad a varios escenarios de detección, mejorando así aún más la precisión. Finalmente, desarrollamos un sistema de detección de conteo de peatones utilizando PyQt5 para mejorar la interacción humano-computadora. Los resultados experimentales en el conjunto de datos público INRIA mostraron que nuestro algoritmo logró una precisión de detección del 98.4%, lo que representa una mejora del 10.1% sobre el algoritmo original YOLOv5s. Este avance mejora significativamente la detección de objetos pequeños en imágenes y aborda de manera efectiva los problemas de detección errónea y omisión en entornos complejos. Estos hallazgos tienen importantes implicaciones prácticas para garantizar la seguridad vial y optimizar el flujo de tráfico.