Detección de humo de incendios mediante fusión de características detalladas y mecanismo de atención híbrido

WANG Rui-qing ,  

WANG Hui-qin ,  

WANG Ke ,  

摘要

Para el problema de la debilitación de las características detalladas en los mapas de características de alto nivel de redes neuronales convolucionales, lo que causa la pérdida de las características de bajo nivel en imágenes de humo, se propone un algoritmo mejorado YOLOv4 que combina características detalladas y un mecanismo de atención híbrido. Se diseñó un módulo de fusión de características detalladas que introduce las características detalladas de bajo nivel de la red principal en el mapa de características de alto nivel, obteniendo características fusionadas con rica información multiescala. Se emplea un mecanismo de atención híbrido en las dimensiones de canal y espacial para reasignar los pesos de las características fusionadas, fortaleciendo las características objetivo del humo mientras se suprimen las características de áreas irrelevantes, haciendo que la expresión de las características del humo sea más robusta. Los resultados experimentales muestran que la precisión promedio, la precisión y la tasa de recuperación del algoritmo propuesto aumentaron un 4.31 %, 1.21 % y 9.86 % respectivamente en comparación con el algoritmo YOLOv4, manteniendo al mismo tiempo una velocidad de detección rápida. Este algoritmo puede extraer eficazmente las características generales del objetivo del humo y es más adecuado para la tarea de detección de humo de incendios en fondos complejos.

关键词

Detección de humo; aprendizaje profundo; algoritmo YOLOv4; fusión de características; mecanismo de atención híbrido

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website