logo móvil
Contáctanos

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

Descargar PDF

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


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.

Otros recursos que podrían interesarte

Temas Virtualpro