半监督单图像去雾
半监督单图像去雾代码。
依赖
pytorch >= 1.0
visdom
数据集制作
通过以下方式使您成为数据集:
合成图像:将两张图像(朦胧(HxWxC),清洁(HxWxC))对齐为一张图像(Hx2WxC)。 要注意的是,H和W应该是8的倍数。将它们( ./datasets/dehazing/train张图像)放在./datasets/dehazing/train 。
真实的模糊图像:将它们( ./datasets/dehazing/unlabeled张图像)放在./datasets/dehazing/unlabeled
测试图像:与1.对齐,然后将它们放在./datasets/dehazing/test
火车
您可以通过以下方式训练模型:
python train.py --dataroot ./datasets/dehazing --name run_
1