Algoritmo de detección de barcos basado en la red YOLOv5 mejorado con convolución ligera y mecanismo de atención
Autores: Wang, Langyu; Zhang, Yan; Lin, Yahong; Yan, Shuai; Xu, Yuanyuan; Sun, Bo
Idioma: Inglés
Editor: MDPI
Año: 2023
Acceso abierto
Artículo científico
2023
Algoritmo de detección de barcos basado en la red YOLOv5 mejorado con convolución ligera y mecanismo de atención
Categoría
Ingeniería y Tecnología
Subcategoría
Ingeniería de Software
Palabras clave
Extracción de características
Precisión
Recordar
Algoritmo YOLOv5
Mecanismo de atención
Convolución ligera
Licencia
CC BY-SA – Atribución – Compartir Igual
Consultas: 31
Citaciones: Sin citaciones
Aiming atacar el problema de extracción de características insuficiente, baja precisión y recall en la detección de barcos en la superficie del mar, se propone un algoritmo YOLOv5 basado en convolución ligera y mecanismo de atención. Combinamos el módulo de mejora del campo receptivo (REF) con el módulo de agrupación rápida de pirámide espacial para retener información semántica más rica y expandir el campo sensorial. Se añade el módulo de cuello delgado basado en una convolución ligera (GSConv) a la sección del cuello, para lograr una mayor eficiencia computacional del detector. Y, para mejorar el rendimiento del modelo y centrarse en la información posicional, se añadió el mecanismo de atención de coordenadas. Finalmente, la función de pérdida CIoU se reemplaza por SIoU. Los resultados experimentales utilizando el conjunto de datos seaShips muestran que en comparación con el algoritmo YOLOv5 original, el algoritmo YOLOv5 mejorado tiene ciertas mejoras en los índices de evaluación del modelo, mientras que el número de parámetros en el modelo no aumenta significativamente, y la velocidad de detección también cumple con los requisitos de detección de barcos en la superficie del mar.
Descripción
Aiming atacar el problema de extracción de características insuficiente, baja precisión y recall en la detección de barcos en la superficie del mar, se propone un algoritmo YOLOv5 basado en convolución ligera y mecanismo de atención. Combinamos el módulo de mejora del campo receptivo (REF) con el módulo de agrupación rápida de pirámide espacial para retener información semántica más rica y expandir el campo sensorial. Se añade el módulo de cuello delgado basado en una convolución ligera (GSConv) a la sección del cuello, para lograr una mayor eficiencia computacional del detector. Y, para mejorar el rendimiento del modelo y centrarse en la información posicional, se añadió el mecanismo de atención de coordenadas. Finalmente, la función de pérdida CIoU se reemplaza por SIoU. Los resultados experimentales utilizando el conjunto de datos seaShips muestran que en comparación con el algoritmo YOLOv5 original, el algoritmo YOLOv5 mejorado tiene ciertas mejoras en los índices de evaluación del modelo, mientras que el número de parámetros en el modelo no aumenta significativamente, y la velocidad de detección también cumple con los requisitos de detección de barcos en la superficie del mar.