【文件属性】:
文件名称:VDNet:变分去噪网络
文件大小:125.71MB
文件格式:ZIP
更新时间:2021-04-29 05:38:57
Python
VDN
要求和依存关系
Ubuntu 16.04,CUDA 10.0
Python 3.6,Pytorch 1.1.0
详细信息(请参阅environment.yml)
图像去噪
非IID高斯噪声消除
在训练阶段,我们使用, 的源图像作为 ,并使用高斯核生成了噪声的方差图。 但是,在测试阶段,采用了三种不同的方差图来生成噪声图像,以验证VDN的一般性。
测验
python demo_test_simulation.py
训练
从以上链接下载源图像。
准备测试数据集:
python datasets/prepare_data/simulation/noise_generate_nips_niid.py
开始培训:
python train_simulation.py --simulate_dir source_imgs_path --eps2 5e-5
真实的