劉飛躍 沈靖夫 楊一帆 許之遠(yuǎn) 張?jiān)?/p>

摘要：針對(duì)船上空間特殊、環(huán)境復(fù)雜的情況，現(xiàn)有的火災(zāi)識(shí)別算法直接應(yīng)用到船上會(huì)出現(xiàn)識(shí)別精度不高的現(xiàn)象。文章在普通的火焰數(shù)據(jù)集的基礎(chǔ)上增加了船上和倉(cāng)庫(kù)起火圖片來(lái)進(jìn)行數(shù)據(jù)集的豐富。文章網(wǎng)絡(luò)模型在YOLOv5s網(wǎng)絡(luò)結(jié)構(gòu)的基礎(chǔ)上進(jìn)行優(yōu)化，修改后的模型對(duì)于火焰檢測(cè)的適應(yīng)性更高。加入了坐標(biāo)注意力（CA）模塊以提高算法對(duì)目標(biāo)的定位能力。將Neck部分的PANet改成了加權(quán)雙向特征金字塔結(jié)構(gòu)（BiFPN） ，以快速地進(jìn)行多尺度特征融合。邊框回歸損失采用了EIOU，以提高回歸精度。改進(jìn)后的網(wǎng)絡(luò)精度增加了6.1%，召回率增加了2.4%，平均精度增加了3%。優(yōu)化后的算法對(duì)于雜亂環(huán)境和畫(huà)質(zhì)不高的情況均有較好的檢測(cè)性能。

關(guān)鍵詞：船；火災(zāi)檢測(cè)；CA；BiFPN

中圖分類號(hào)：TP311 文獻(xiàn)標(biāo)識(shí)碼：A文章編號(hào)：1009-3044（2023）29-0019-04