Détection de fumée d'incendie par fusion de caractéristiques détaillées et mécanisme d'attention hybride

WANG Rui-qing ,  

WANG Hui-qin ,  

WANG Ke ,  

摘要

Pour le problème de l'affaiblissement des caractéristiques détaillées dans les cartes de caractéristiques de haut niveau des réseaux neuronaux convolutifs, entraînant une perte des caractéristiques de bas niveau dans les images de fumée, une méthode améliorée YOLOv4 intégrant des caractéristiques détaillées et un mécanisme d'attention hybride est proposée. Un module de fusion des caractéristiques détaillées a été conçu pour introduire les caractéristiques détaillées de bas niveau du réseau principal dans la carte des caractéristiques de haut niveau, obtenant ainsi des caractéristiques fusionnées riches en informations multi-échelles. Un mécanisme d'attention hybride est utilisé sur les dimensions de canal et spatiale pour réattribuer les poids des caractéristiques fusionnées, renforçant les caractéristiques des cibles de fumée tout en supprimant les caractéristiques des zones non pertinentes, ce qui rend l'expression des caractéristiques de fumée plus robuste. Les résultats expérimentaux montrent que la précision moyenne, la précision et le rappel de l'algorithme proposé ont augmenté respectivement de 4,31 %, 1,21 % et 9,86 % par rapport à l'algorithme YOLOv4, tout en maintenant une vitesse de détection rapide. Cet algorithme peut extraire efficacement les caractéristiques globales des cibles de fumée et est plus adapté à la détection de fumée d'incendie dans des contextes complexes.

关键词

Détection de fumée; apprentissage profond; algorithme YOLOv4; fusion de caractéristiques; mécanisme d'attention hybride

阅读全文

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

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