Reconstrução de super-resolução de imagem baseada em rede de atenção multidimensional

HE Xing ,  

WANG Lei ,  

ZHANG Pengchao ,  

WANG Shusheng ,  

ZHANG Heng ,  

摘要

Os métodos atuais de reconstrução de super-resolução de imagem única baseados em modelos probabilísticos de difusão apresentam insuficiências na extração de informações espaciais, não exploram suficientemente as informações relevantes e apresentam redundâncias no processo computacional. Este artigo projeta um método de reconstrução de super-resolução de imagem única que incorpora uma rede de atenção multidimensional. Primeiramente, com base no modelo de difusão SRDiff, é proposta uma atenção multidimensional que combina atenção de canal, autoatenção e atenção espacial, aumentando a capacidade do modelo de capturar características em diferentes escalas, permitindo preservar mais detalhes e melhor consistência global ao restaurar imagens de alta resolução. Em seguida, é introduzida a convolução parcial (PConv) para extrair com precisão as características espaciais da imagem, melhorando a qualidade do resultado da super-resolução e reduzindo significativamente a carga computacional, aumentando assim a eficiência do modelo. Sob um fator de ampliação de 4, o método foi comparado a outros em 5 conjuntos de teste, mostrando resultados com um aumento médio de 0,762 dB no PSNR e 0,082 na similaridade estrutural em comparação a outros métodos. O método proposto apresenta subjetivamente mais detalhes finos e melhores efeitos visuais, e objetivamente valores mais altos de PSNR e similaridade estrutural.

关键词

super-resolução; modelo de difusão; rede de atenção multidimensional; convolução parcial

阅读全文

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

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