基于圖像增強(qiáng)和YOLO的船舶火災(zāi)檢測(cè)方法
上海海事大學(xué)學(xué)報(bào)
頁(yè)數(shù): 7 2024-06-30
摘要: 為提高船舶火災(zāi)檢測(cè)的響應(yīng)速度,針對(duì)海洋環(huán)境下船舶火災(zāi)的特點(diǎn),提出一種基于圖像增強(qiáng)算法和YOLO算法的海上船舶火災(zāi)檢測(cè)方法。構(gòu)建一套面向海上船舶火災(zāi)的數(shù)據(jù)集,對(duì)數(shù)據(jù)集圖像進(jìn)行增強(qiáng)處理,以突出火焰細(xì)節(jié)。對(duì)網(wǎng)絡(luò)的頭部結(jié)構(gòu)進(jìn)行改進(jìn),只保留其中的特征金字塔網(wǎng)絡(luò)(feature pyramid network, FPN)結(jié)構(gòu),在保證模型精度的前提下減少模型參數(shù),加快計(jì)算速度。通過(guò)實(shí)驗(yàn)進(jìn)行驗(yàn)...