1.一種基于混合注意力編碼器的圖像去霧方法,其特征在于:采用混合注意力編碼器對(duì)輸入的霧圖進(jìn)行特征提取;然后對(duì)提取的霧圖特征與離散碼本先驗(yàn)進(jìn)行匹配;之后,對(duì)匹配后的特征進(jìn)行解碼;最后,對(duì)解碼器特征進(jìn)行特征嵌入,逐步解碼獲得無(wú)霧結(jié)果。
2.根據(jù)權(quán)利要求1所述的基于一種基于混合注意力編碼器的圖像去霧方法,其特征在于:所述混合注意力編碼器基于transformer和通道注意力機(jī)制,在swin-transformer的基礎(chǔ)結(jié)構(gòu)上引入channel?attention,采用由通道注意力模塊和窗口多頭自注意力模塊組成的并聯(lián)注意力模塊pab,以替換窗口多頭自注意力模塊。
3.根據(jù)權(quán)利要求1所述的一種基于混合注意力編碼器的圖像去霧方法,其特征在于:獲取圖像后,使用所述混合注意力編碼器進(jìn)行多尺度的特征提取以得到最終的編碼器特征:對(duì)于給定的圖像輸入h、w、ch分別表示輸入的長(zhǎng)、寬、通道數(shù);首先利用一個(gè)卷積層和若干殘差層進(jìn)行淺層特征提取,得到cs表示中間特征的通道數(shù);然后,經(jīng)過(guò)由四個(gè)殘差混合注意力組模塊rhag組成的深層特征提取模塊,得到編碼器特征其中,每個(gè)殘差混合注意力組模塊rhag都包含六個(gè)混合注意力基本模塊hab、一個(gè)重疊交叉注意力機(jī)制,一個(gè)卷積層。
4.根據(jù)權(quán)利要求3所述的一種基于混合注意力編碼器的圖像去霧方法,其特征在于:
5.根據(jù)權(quán)利要求4所述的一種基于混合注意力編碼器的圖像去霧方法,其特征在于:在每個(gè)所述殘差混合注意力組模塊中引入一個(gè)重疊交叉注意力機(jī)制。
6.根據(jù)權(quán)利要求1所述的一種基于混合注意力編碼器的圖像去霧方法,其特征在于:得到編碼器特征之后,采用二元匹配機(jī)制使特征與離散碼本先驗(yàn)進(jìn)行匹配,通過(guò)調(diào)整具有二次特性的調(diào)制曲線的參數(shù),賦予匹配機(jī)制在不同去霧場(chǎng)景下更大的可變性。
7.根據(jù)權(quán)利要求6所述的一種基于混合注意力編碼器的圖像去霧方法,其特征在于:
8.根據(jù)權(quán)利要求7所述的一種基于混合注意力編碼器的圖像去霧方法,其特征在于:
9.根據(jù)權(quán)利要求1所述的一種基于混合注意力編碼器的圖像去霧方法,其特征在于:在特征提取和二元匹配之后,獲得編碼器特征fh以及對(duì)應(yīng)的與codebook匹配后的特征f'h;之后,將fh、f'h分別輸入解碼器d和vqgan的解碼器dc中,分別得到對(duì)應(yīng)的解碼器特征f1和f2;采用帶有可變形卷積的特征嵌入模塊fem將解碼器特征f2嵌入f1,以恢復(fù)無(wú)霧的清晰結(jié)果。
10.一種基于混合注意力編碼器的圖像去霧系統(tǒng),其特征在于:包括存儲(chǔ)器、處理器及存儲(chǔ)在存儲(chǔ)器上并可在處理器上運(yùn)行的計(jì)算機(jī)程序,其特征在于,所述處理器執(zhí)行所述程序時(shí)實(shí)現(xiàn)如權(quán)利要求1-9任一項(xiàng)所述的方法。