基于GAN的图像去雾
该项目旨在实现基于GAN的图像去雾方法
图1.生成器。 发生器可以直接输出无雾图像,而无需估计中间参数。
图2.鉴别器。 鉴别器可以引导发生器生成更真实的除雾结果。
要运行此项目,您需要设置环境,下载数据集,然后可以训练和测试网络模型。 我将逐步向您展示运行该项目,希望它足够清楚。
先决条件
该项目已在Ubuntu 16.04的GPU Titan V上进行了测试。请注意,运行该代码需要一个GPU。 否则,您必须稍微修改一些代码才能使用CPU。 如果使用CPU进行训练,可能会太慢。 因此,我建议您使用足够强大的GPU和大约12G的RAM。
依存关系
建议使用Python 3.5或3.6。
tqdm==4.19.9
numpy==1.17.3
tensorflow==1.12.0
tensorboardX
torch==1.0.0
Pillow==6.2.0
to
2021-11-10 20:18:26
16.92MB
Python
1