![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/Architecture.jpg)
Unet1 |
Unet2 |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/Unet.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/Unet2.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/discriminator.jpg)
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/training epochs.jpg)
python main.py --phase train --pretrain (True or False) --weights_path ./weights_dir/weights.ckpt --dataset_path_x ./dataset_x.npy --dataset_path_y ./dataset_y.npy --output_dir ./output/
python main.py --phase test --pretrain True --weights_path ./weights_dir/weights.ckpt --dataset_path_x ./dataset_x.npy --dataset_path_y ./dataset_y.npy --output_dir ./output/
python main.py --phase images --pretrain True --weights_path ./weights_dir/weights.ckpt --imlist ["./sample1.jpg","./sample2.png"] --output_dir ./output/
python main.py --phase video --pretrain True --weights_path ./weights_dir/weights.ckpt --video_path ./video.avi --output_dir ./output/
input |
output |
input |
output |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/input1.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/output1.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/input1.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/output1.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/input2.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/output2.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/input2.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/output2.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/input3.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/output3.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/input3.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/output3.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/input4.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/output4.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/input4.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/output4.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/input5.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/output5.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/input5.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/output5.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/input6.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/land/output6.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/input6.jpg) |
![](https://raw.githubusercontent.com/colalon/dehaze-GAN/master/./doc/underwater/output6.jpg) |