logo móvil
Contáctanos

Kpe-yolov5: un algoritmo mejorado de detección de objetivos pequeños basado en yolov5

Autores: Yang, Rujin; Li, Wenfa; Shang, Xinna; Zhu, Deping; Man, Xunyu

Idioma: Inglés

Editor: MDPI

Año: 2023

Descargar PDF

Acceso abierto

Artículo científico
2023

Kpe-yolov5: un algoritmo mejorado de detección de objetivos pequeños basado en yolov5


Categoría

Ingeniería y Tecnología

Subcategoría

Ingeniería Eléctrica y Electrónica

Palabras clave

Propone
Algoritmo kpe-yolov5
Detección de objetivos pequeños
Precisión
Módulo de atención scse
Extracción de características
Conjunto de datos visdrone-2020

Licencia

CC BY-SA – Atribución – Compartir Igual

Consultas: 28

Citaciones: Sin citaciones


Descripción
En la actualidad, los métodos existentes tienen muchas limitaciones en la detección de objetivos pequeños, como la baja precisión, una alta tasa de detección falsa y la detección perdida. Este documento propone el algoritmo KPE-YOLOv5 con el objetivo de mejorar la capacidad de detección de objetivos pequeños. El algoritmo tiene tres mejoras basadas en el algoritmo YOLOv5. En primer lugar, logra un tamaño más preciso de las cajas de anclaje para los objetivos pequeños mediante la tecnología de agrupamiento K-means++. En segundo lugar, el módulo de atención scSE (compresión y excitación espacial y de canal) se integra en el nuevo algoritmo para fomentar que la red base preste mayor atención a la información de características de los objetivos pequeños. Por último, se mejora la capacidad de extracción de características de objetivos pequeños aumentando la capa de detección de objetivos pequeños, lo que también aumenta la precisión de detección de objetivos pequeños. Evaluamos KPE-YOLOv5 en el conjunto de datos VisDrone-2020 y comparamos el rendimiento con YOLOv5. Los resultados muestran que KPE-YOLOv5 mejora el mAP de detección en un 5.3% y aumenta el P en un 7%. El algoritmo KPE-YOLOv5 tiene un mejor resultado de detección que YOLOv5 para la detección de objetivos pequeños.

Otros recursos que podrían interesarte

Temas Virtualpro