zhengziqiang / ForkGAN

ForkGAN: Seeing into the rainy night. ECCV 2020 (oral). A task-agnostic image translation framework that can boost multiple vision tasks in adverse weather conditions, including localization, semantic segmentation and object detection.
121 stars 16 forks source link

PyTorch implementation of ForkGAN #14

Open seawee1 opened 3 years ago

seawee1 commented 3 years ago

Hi, FYI, I created a PyTorch implementation of ForkGAN (https://github.com/seawee1/ForkGAN-pytorch.git). It's based on the CycleGAN repository and features some neat little extras (MultiGPU training, instance-level losses, automatic mixed precision for smaller GPUs).