/imageDenoise-UNet-VGG

拓片图像去噪，以UNet为基本框架，编码器基于VGG16

Primary LanguagePython

imageDenoiseUsingUNet-VGG

拓片图像去噪网络，以UNet为基本框架，编码器基于VGG16

环境：

Python 3.5.3
Pytorch 1.2.0

使用方法：

1、准备成对的数据集，以相同的名字分别放置在data/src和data/target文件夹下
2、如果图像的尺寸过大&&电脑内存不充足，可利用data/文件夹下的脚本进行剪切处理
3、使用train.py进行训练

实验结果：

数据集：甲骨拓片数据集。
对比网络：CycleGAN
UNet-VGG的效果优于CycleGAN，可以看出CycleGAN对于字体有弱化现象，且对于细节的处理没有UNet-VGG好。

CycleGAN处理效果

image

UNet-VGG处理效果

image