Pytorch图像去噪实战(二十四):批量图片去噪脚本实战,构建可复用的数据处理流水线
Pytorch图像去噪实战(二十四):批量图片去噪脚本实战,构建可复用的数据处理流水线
一、问题场景:一张图能处理,几万张图怎么办?
前面我们已经实现了单张图片去噪、服务部署、大图分块推理。
但真实项目里,经常不是处理一张图,而是:
- 一批OCR图片
- 一批商品图
- 一批扫描件
- 一批监控帧
- 几万张历史图片
如果每次手动调用脚本处理一张图,效率非常低。
所以这一篇我们要解决:
如何写一个可复用的批量图片去噪脚本。
二、批处理脚本应该具备哪些能力?
一个工程可用的批处理脚本,不应该只是简单 for 循环。
至少要支持:
- 输入目录
- 输出目录
- 自动遍历图片
- 保持文件名
- 支持异常跳过
- 显示进度
- 支持GPU/CPU
- 支持断点续跑
- 保存日志
这才是可以长期使用的工具。
三、工程目录结构
batch_denoise/ ├── models/ │ └── une