Rede de aprimoramento de imagens subaquáticas com fusão global de características baseada numa U-Net aprimorada

GAO Shaoshu ,  

JIAO Guangsen ,  

LI Guangfeng ,  

LIU Zongen ,  

摘要

Propõe-se uma rede de aprimoramento de imagens subaquáticas com fusão global de características baseada numa U-Net aprimorada para abordar o problema de desvios de cor e desfoque de detalhes causados pela dispersão e atenuação da luz durante a propagação em ambiente subaquático. Primeiramente, módulos convolucionais residuais múltiplos foram projetados no codificador e decodificador para realizar a fusão hierárquica das informações de características, reduzindo a perda de detalhes. Em seguida, um módulo de atenção por canal foi introduzido no decodificador para ponderar os canais, aliviando o problema da degradação desigual dos canais. Por fim, um módulo de autoatenção convolucional com permutação foi projetado no decodificador para integrar informações globais, promovendo a reconstrução de imagens guiada pela rede. O método proposto foi testado no conjunto de dados UIEB, alcançando resultados de 23,42, 0,9005 e 0,1385 nos índices PSNR, SSIM e LPIPS, respectivamente, e no conjunto de dados LSUI, alcançando 29,35, 0,9382 e 0,0880 nos mesmos índices. Os resultados experimentais demonstram que o método proposto é eficaz e viável para restaurar desvios de cor e reduzir o desfoque de detalhes.

关键词

Aprimoramento de imagens subaquáticas; aprendizado profundo; fusão de características; mecanismo de atenção; redes neurais convolucionais

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website