TensorFlow实现的深度CNN图像去噪模型DnCNN:超越高斯denoiser
需积分: 11 149 浏览量
更新于2024-09-04
收藏 3KB MD 举报
DnCNN-tensorflow 是一个基于 TensorFlow 实现的深度卷积神经网络(Deep Convolutional Neural Network, DCNN)模型,用于图像去噪。该模型是 2017 年《超越高斯去噪器:深度卷积神经网络的残差学习》(Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising) 一文的实现,论文作者是张驰等人。这个项目包含了模型架构图以及在 BSD68 和 Set12 数据集上的性能对比。
模型架构部分展示了 DnCNN 的具体结构,通过使用残差学习(residual learning),模型能够学习到更复杂的图像特征并进行有效的降噪。模型的图形化表示位于 `./img/model.png` 文件中,展示了网络的层级和连接方式。
在结果部分,首先展示了在 BSD68 数据集上,不同方法在噪声级别为 25 dB 的情况下,平均峰值信噪比(Peak Signal-to-Noise Ratio, PSNR)得分。DnCNN-tensorflow 在这个测试中表现优秀,超过了其他如 BM3D、WNNM、EPLL、MLP、CSF 和 TNRD 等经典去噪算法。而在 Set12 数据集的测试中,尽管没有具体列出 DnCNN-S 的分数,但 DnCNN-tensorflow 的得分略高于 DnCNN-S。
该项目还提供了两种运行环境的选项:推荐的 Docker 集成环境,可以方便地利用 GPU 资源,并且提供了预训练模型的下载链接。对于不使用 Docker 的情况,需要安装 TensorFlow 1.4.1、NumPy 等依赖库。项目还包括了一个“一键去噪”脚本 `./oneKeyToDenoise.sh`,用户可以通过这个脚本在测试集上快速处理图像去噪。
训练过程需要执行 `generate_patches.py` 和 `main.py`,并支持命令行参数调整。原始模型的训练大约需要 4 小时在 GTX 1080TI 上。此外,项目还展示了训练过程中损失函数的变化图,但提示这张图可能不再适合此模型。
待办事项列表列出了项目的后续改进计划,包括修复已知问题、代码优化、使用 Docker 进行一键去噪、与原版 DnCNN 对比、更新库依赖等。最后,项目对参与贡献者表示感谢。
DnCNN-tensorflow 是一个实用的图像去噪工具,适用于研究和实际应用中,并且提供了一个可扩展和优化的基础框架。
点击了解资源详情
点击了解资源详情
点击了解资源详情
2021-02-04 上传
2021-02-04 上传
2021-05-14 上传
hank黄
- 粉丝: 1
- 资源: 1
最新资源
- 火炬连体网络在MNIST的2D嵌入实现示例
- Angular插件增强Application Insights JavaScript SDK功能
- 实时三维重建:InfiniTAM的ros驱动应用
- Spring与Mybatis整合的配置与实践
- Vozy前端技术测试深入体验与模板参考
- React应用实现语音转文字功能介绍
- PHPMailer-6.6.4: PHP邮件收发类库的详细介绍
- Felineboard:为猫主人设计的交互式仪表板
- PGRFileManager:功能强大的开源Ajax文件管理器
- Pytest-Html定制测试报告与源代码封装教程
- Angular开发与部署指南:从创建到测试
- BASIC-BINARY-IPC系统:进程间通信的非阻塞接口
- LTK3D: Common Lisp中的基础3D图形实现
- Timer-Counter-Lister:官方源代码及更新发布
- Galaxia REST API:面向地球问题的解决方案
- Node.js模块:随机动物实例教程与源码解析