基于PyTorch的DnCNN去噪代码Matlab-DnCNN-PyTorch实现

这份代码使用PyTorch实现基于Matlab的DnCNN去噪算法。运行此代码需要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、Wireless-Wavelet、Lock Phase Loop、MLP、SpineNet、TNRD、Neural Networks以及改进的DnCNN-S-Re，其中DnCNN-S-Re的PSNR达到了31.75dB。