unet 实现文本文档去噪、去水印
torch 1.2.0
torchvision 0.4.0
opencv-python 4.1.0.25
numpy 1.16.2
下载链接:denoising-dirty-documents数据集
提取码:d8l7
python train.py
数据增强在 transforms.py 包括灰度化、水平垂直翻转、旋转、裁剪、直方图均衡化、均值滤波、Log灰度变换
python predict.py
下载链接:预训练模型1.5M
提取码:poyf