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