融合外部注意力的擴(kuò)散模型巡檢圖像去霧
摘要: 為降低霧天對輸電線路巡檢圖像造成的影響,針對目前主流去霧算法存在計(jì)算成本高、圖像去霧后檢測性能差和難以部署的問題,提出了一種霧天輸電線路巡檢圖像去霧方法Diff-EaT。該方法采用融合Transformer的擴(kuò)散模型結(jié)構(gòu),為降低體征提取模塊中多頭自注意力在ViT中的計(jì)算復(fù)雜度,使用多頭外部注意力代替多頭自注意力以減少計(jì)算負(fù)荷和增強(qiáng)特征學(xué)習(xí)。同時(shí),設(shè)計(jì)了一個(gè)混合尺度門控前饋網(wǎng)絡(luò),... (共9頁)
開通會員,享受整站包年服務(wù)