面向多天气退化图像恢复的自注意力扩散模型

秦菁, 文渊博, 高涛, 刘瑶

A Transformer-Based Diffusion Model for All-in-One Weather-Degraded Image Restoration

QIN Jing, WEN Yuanbo, GAO Tao, LIU Yao

表1 不同自注意力方法的计算量对比

Tab.1 Comparison of computational amount of different self-attention methods

自注意力方法 乘法计算量 加法计算量 SA 1.07×109 1.06×109 TSA 8.39×106 8.26×106 STSA 2.62×106 2.49×106