Keyboard shortcuts

Press or to navigate between chapters

Press S or / to search in the book

Press ? to show this help

Press Esc to hide this help

DruNet ncnn Windows

项目概览

这是一个基于 ncnn 的 DRUNet 非官方简化实现,目标是把图像去噪模型整理成更轻量、也更适合 Windows 环境部署的形式。

从当前仓库内容来看,这个项目已经不只是“把模型跑通”,而是开始进入工程整理阶段:依赖环境、调用方式和一组基础性能数据都已经明确下来,适合作为后面继续补部署细节和 benchmark 的基础版本。

环境与使用方式

当前项目依赖环境主要包括:

  • Visual Studio 2019 / 2022
  • NCNN 预编译版本
  • OpenCV 4.6.0

编译完成后,当前的调用方式比较直接:

./<xxx.exe> <image-path>

这也说明项目当前重点更偏向推理落地,而不是封装成复杂的上层接口。

结果与性能

项目说明里已经给出一组基础运行时间,测试平台是 AMD Ryzen 5 5600G,目前公开的数据主要集中在 Vulkan 推理:

分辨率核显 + Vulkan
256x2562.62
1200x160019.73

这组结果虽然还不算完整,但已经足够作为后续继续补 CPU、独显和更多输入尺寸测试的基线。

项目说明中还提到了灰度图和彩色图的结果对照,不过当前仓库公开路径下没有对应的结果图片文件,因此这里先不放失效图片,后面如果仓库补齐资源,可以直接把对比图接进来。

后续值得补充的方向

  • 模型转换和推理流程
  • Windows 端部署细节
  • 速度、显存和效果对比
  • 与原始 Pytorch 输出的一致性说明

参考

  • Zhang et al., Plug-and-Play Image Restoration with Deep Denoiser Prior, TPAMI 2021

技术栈

  • C++
  • ncnn
  • Denoising

链接