1024210879/unet-denoising-dirty-documents

参考眼底图像血管分割retina-unet实现的文档去噪、去水印

unet-denoising-dirty-documents

unet 实现文本文档去噪、去水印

requirement

torch 1.2.0
torchvision 0.4.0
opencv-python 4.1.0.25
numpy 1.16.2

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

reference

眼底图像血管分割 retina-unet