logo móvil
Contáctanos

Un modelo mejorado de YOLOv5s para la detección de edificios

Autores: Zhao, Jingyi; Li, Yifan; Cao, Jing; Gu, Yutai; Wu, Yuanze; Chen, Chong; Wang, Yingying

Idioma: Inglés

Editor: MDPI

Año: 2024

Descargar PDF

Acceso abierto

Artículo científico
2024

Un modelo mejorado de YOLOv5s para la detección de edificios


Categoría

Ingeniería y Tecnología

Subcategoría

Ingeniería Eléctrica y Electrónica

Palabras clave

Tecnología de vehículos autónomos
Reconocimiento de edificios
Dispositivos de borde
Modelo ligero
YOLOv5s
Oclusión

Licencia

CC BY-SA – Atribución – Compartir Igual

Consultas: 60

Citaciones: Sin citaciones


Descripción
Con el avance continuo de la tecnología de vehículos autónomos, el reconocimiento de edificios se vuelve cada vez más crucial. Permite a los vehículos autónomos comprender mejor su entorno circundante, facilitando una navegación más segura y procesos de toma de decisiones. Por lo tanto, es importante mejorar la eficiencia de detección en dispositivos periféricos. Sin embargo, el reconocimiento de edificios enfrenta problemas como la severa oclusión y el gran tamaño de los modelos de detección que no pueden ser desplegados en dispositivos periféricos. Para resolver estos problemas, se propone en este estudio un modelo ligero de reconocimiento de edificios basado en YOLOv5s. Primero recopilamos un conjunto de datos de edificios de escenas reales e internet, y aplicamos un método de aumento de datos GridMask mejorado para expandir el conjunto de datos y reducir el impacto de la oclusión. Para hacer que el modelo sea ligero, podamos el modelo mediante el método de poda de canales, lo que disminuye los costos computacionales del modelo. Además, utilizamos como función de activación para ayudar al modelo a converger mejor en el entrenamiento disperso. Finalmente, comparándolo con YOLOv5s (referencia), los experimentos muestran que el modelo mejorado reduce el tamaño del modelo en 9.595 MB, y el mAP@0.5 alcanza el 82.3%. Este estudio ofrecerá información sobre la detección ligera de edificios, demostrando su importancia en la percepción ambiental, monitoreo y detección, especialmente en el campo de la conducción autónoma.

Otros recursos que podrían interesarte

Temas Virtualpro