雨痕退化预测与预训练扩散先验的单图像去雨方法.pdfVIP

3042025，61（6）ComputerEngineeringandApplications计算机工程与应用

雨痕退化预测与预训练扩散先验的单图像去雨方法

+

谢瑞麟，吴昊，袁国武

云南大学信息学院，昆明650504

+通信作者E-mail：haowu_sise@ynu.edu.cn

摘要：雨天拍摄的图像由于雨痕干扰，会降低视觉质量和后续任务精度。为有效应用扩散模型中的生成式先验以

及避免重新训练条件扩散模型带来的计算负担，提出了一种结合雨痕退化预测与无条件预训练扩散模型的单图像

去雨方法，通过使用卷积字典学习机制在雨痕退化预测网络中获取带雨图像的雨痕图，将雨痕图用于引导零空间扩

散模型，实现了使用在已有的预训练无条件扩散模型下进行单图像去雨，从而有效地提高了图像去雨的质量。和其

他单图像去雨方法相比，该方法在Rain100H和Rain100L数据集上取得了目前最好的结果，PSNR指标最大提升了

0.44dB（+1.1%），SSIM指标最大提升了0.006（+0.7%），LPIPS指标最大提升了0.008（+42.1%）。

关键词：单图像去雨；扩散模型；预训练模型

文献标志码：A中图分类号：TP391.41doi：10.3778/j.issn.1002-8331.2311-0264

SingleImageDerainingUsingRainyStreakDegradationPredictionandPre-TrainedDiffusionPrior

+

XIERuilin,WUHao,YUANGuowu

SchoolofInformationScienceandEngineering,YunnanUniversity,Kunming650504,China

Abstract：Imagescapturedinrainyweatherdegradevisualqualityandsubsequenttaskaccuracyduetointerferencefrom

rainstreaks.Toeffectivelyapplythegenerativepriorinthediffusionmodelaswellastoavoidthecomputationalburden

ofre-trainingtheconditionaldiffusionmodel,asingleimagederainingmethodcombiningrainstreakdegradationpredic-

tionandunconditionalpre-traineddiffusionmodelisproposed,whichisachievedbyusingtheconvolutionaldictionary

learningmechanismtoobtainrainstreakmapswithrainingimages,therainstreakmapwillbeusedtobootstrapthenull-

spacediffusionmodel.Singleimagederainingusinganexistingpre-trainedunconditionaldiffusionmodel,thuseffectively

improvingthequalityofimagederaining.Comparedwithothersingleimagederainingmethods,themethodachievesthe

highestPSNRimprovementof0.44dB(+1.1%),SSIMimprovementof0.006(+0.7%),andLPIPSimprovementof0.008

(+42.1%)ontheRain100HandRain100Ldatasets,achievingthestate-of-the-artresultssofaronbothdatasets.

Keyw