TensorFlow and PyTorch implementations of the paper Fast Underwater Image Enhancement for Improved Visual Perception (RA-L 2020) and other GAN-based models.
- Training pipelines for FUnIE-GAN and UGAN (original repo) on TensorFlow (Keras) and PyTorch
- Modules for image quality analysis based on UIQM, SSIM, and PSNR (see Evaluation)
Enhanced underwater imagery | Improved detection and pose estimation |
---|---|