Pytorch implemation a real time raindrop model based on GAN.A extremely lightly model to make video real time raindrop.
Prerequisites:
- pytorch 3.6
- pytorch 0.4.1
- ubuntu 16.04
Test on Image
python test_image.py --input input_image --output output_image
Test on Video
python test_video.py --input video.mp4
Results
References
- Non-locally Enhanced Encoder-Decoder Network for Single Image De-raining. Guanbin Li, Xiang He, Wei Zhang, Huiyou Chang, Le Dong, Liang Lin.
- Generative Adversarial Networks. Ian J. Goodfellow, Jean Pouget-Abadie, Mehdi Mirza, Bing Xu, David Warde-Farley, Sherjil Ozair, Aaron Courville, Yoshua Bengio.
- YOLOv3:An Incremental Improvement. Joseph Redmon, Ali Farhadi.