Unet实现文档图像去噪、去水印

requirement

torch 1.2.0 torchvision 0.4.0 opencv-python 4.1.0.25 numpy 1.16.2

代码 以及运行教程 获取：

关注微信公众号 datayx 然后回复 去噪 即可获取。

AI项目体验地址 https://loveai.tech

usage

1. dataset

下载链接：denoising-dirty-documents数据集 提取码：d8l7

2. train

python train.py 数据增强在 transforms.py 包括灰度化、水平垂直翻转、旋转、裁剪、直方图均衡化、均值滤波、Log灰度变换

3. predict

python predict.py

4. pretrained model

下载链接：预训练模型1.5M 提取码：poyf

test images