文章:
NODE-ImgNet: A PDE-informed effective and robust model for image denoising
代码链接:
基于NODE结构和残差结构的关系,利用PDE去噪模型的一般框架,结合基本CNN去噪模型结构设计了基于PDE的图像去噪模型。模型的优势在于参数量少,并且根据时间步长的值可以灵活调整模型的复杂度。
模型的主要框架:
训练数据集:Waterloo Exploration Database
patch_size=60
batch_size=40
epoch=50
测试了灰度图、彩色图以及真实图像去噪实验,与基于CNN的模型进行了比较。