基于Swin Transformer和PJF顏色空間的火災(zāi)檢測算法

打開文本圖片集

摘  要： 為了更加準(zhǔn)確和可靠地在監(jiān)控圖像中對火災(zāi)進(jìn)行早期監(jiān)測，以便輔助火焰檢測儀器更加高效和準(zhǔn)確地檢測火焰，文中基于Transformer算法和PJF顏色空間，設(shè)計(jì)了一種深度學(xué)習(xí)的火災(zāi)圖像檢測算法。首先，對數(shù)據(jù)集中的火災(zāi)圖像數(shù)據(jù)進(jìn)行預(yù)處理，將圖像從RGB顏色空間轉(zhuǎn)化為PJF顏色空間，突出顯示圖像的火焰區(qū)域；然后，將預(yù)處理后的圖像輸入到Swin Transformer網(wǎng)絡(luò)中，過程中采用注意力機(jī)制、移動窗口機(jī)制、相對位置偏移機(jī)制對火災(zāi)圖像進(jìn)行特征提取；最后，將提取的圖像特征送入全連接層，獲取火災(zāi)圖像的檢測結(jié)果。（剩余9509字）