去噪声代码matlab
DnCNN-火炬
基于pytorch的工具
运行此代码
具有依赖项的Python3:scipy,numpy,scikit-image,PIL,h5py,pytorch>
=
0.4
生成训练数据
'generate_data.py'。
您可能需要修改训练数据集的路径。
根据作者提供的信息,生成的训练数据集中有一些空白(零)数据。
训练
[火车]“
train_DnCNN.py”
[模型]'model_DnCNN_
*'包括使用我的函数生成的数据集训练的模型。
测验
[测试]'ValidateResult_DnCNN.py'您需要更改测试集的路径和文件类型。
虽然我通过matlab生成了测试集,以在与其他方法竞争时发挥作用。
结果
在我的工作中,只采用了降噪方法。
高斯去噪
BSD68数据集上不同方法的平均PSNR(dB)结果。
噪音等级
BM3D
无线网络
锁相环
MLP
脑脊液
TNRD
神经网络
神经网络
DnCNN-S-Re
15
31.07
31.37
31.21
--
31.24
31.42
31.73
31.61
31.75
25
28.57
28.
2021-11-22 15:05:15
14.93MB
系统开源
1