融合注意力機制的輕量級火災檢測模型
消防科學與技術
頁數(shù): 6 2024-03-15
摘要: 基于視覺信息的火災檢測對消防工作具有重要意義,但現(xiàn)階段相關研究提出的方法大多是基于高性能的硬件設備開展,這限制了相關成果的實際應用。在YOLOv5目標檢測算法基礎上使用ShuffleNetv2網(wǎng)絡為主干構造輕量化模型,并引入SIoU損失函數(shù)提高模型目標框的定位精度,同時在模型中添加Shuffle Attention注意力機制,提高在復雜環(huán)境下對目標火焰的識別精度。試驗證明,與Y... (共6頁)