自导网络快速图像去噪
SGN的PyTorch实现以及给定噪声范围的估计PSNR
训练
我在Python 3.6和PyTorch 1.0环境上训练了此SGN。 培训策略与论文相同。 您可以使用以下脚本对自己的数据进行训练(请注意,您需要修改数据集路径):
cd SGN
python train.py or sh zyz.sh
测验
我使用ILSVRC2012验证集对4个NVIDIA TITAN Xp GPU进行了培训,并在1个TITAN Xp GPU上进行了测试。 详细信息显示在代码train.py 。 该演示来自SGN的ILSVRC2012验证集(mu = 0,sigma = 30,batchsize = 32、1000000次迭代)。
左:干净的图像(从COCO2014验证集中选择,COCO_val2014_000000264615.jpg)
中:加性高斯噪声+清晰图像
右
2022-12-15 22:52:09
2.47MB
Python
1