LANet

Our work proposes an adaptive learning attention network (LANet) to solve the problem of color casts and low illumination in underwater images.

Dependencies

The code runs with Python=3.6 and requires Pytorch of version 1.7 or higher. Please pip install the following packages:

numpy=1.20.2
torchvision=0.8.0
matplotlib=3.4.2
opencv-python=4.5.2.54
scipy=1.7.0

Training

1. Download the code
2. run Python train.py --input_images-path ./data/trainA/ --label_images_path ./data/trainB/ 
3. Find checkpoint in the "./checkpoints/" folder
The training data includes input data and label data. input data are in the "./data/trainA" folder, label data are in the "./data/trainB" folder

Testing

1. pre-trained models in the "./checkpoints/" folder
2. Put your testing images in the "./data/test/" folder 
3. run Python test.py --test_pth ./data/test/ --snapshot_pth ./checkpoints/model_epoch_40.pk
4. Find the result in "./results" folder

Contact

If you have any questions, please contact Shiben Liu at [email protected].

Underwater image enhancement

Related tags

Overview

LANet

Dependencies

Training

Testing

Contact

Owner

LiuShiBen

Code Release for Learning to Adapt to Evolving Domains

TensorFlow implementation of Barlow Twins (Barlow Twins: Self-Supervised Learning via Redundancy Reduction)

Text Generation by Learning from Demonstrations

Detection of drones using their thermal signatures from thermal camera through YOLO-V3 based CNN with modifications to encapsulate drone motion

TransMorph: Transformer for Medical Image Registration

Non-Official Pytorch implementation of "Face Identity Disentanglement via Latent Space Mapping" https://arxiv.org/abs/2005.07728 Using StyleGAN2 instead of StyleGAN

Data-driven reduced order modeling for nonlinear dynamical systems

cl;asification problem using classification models in supervised learning

Source code of our BMVC 2021 paper: AniFormer: Data-driven 3D Animation with Transformer

Code for ICCV 2021 paper Graph-to-3D: End-to-End Generation and Manipulation of 3D Scenes using Scene Graphs

An algorithm study of the 6th iOS 10 set of Boost Camp Web Mobile

This repository contains the code for the CVPR 2020 paper "Differentiable Volumetric Rendering: Learning Implicit 3D Representations without 3D Supervision"

A simple pygame dino game which can also be trained and played by a NEAT KI

Gym Threat Defense

Using pretrained language models for biomedical knowledge graph completion.

😊 Python module for face feature changing

A library that can print Python objects in human readable format

Self-Supervised Collision Handling via Generative 3D Garment Models for Virtual Try-On

Generating Anime Images by Implementing Deep Convolutional Generative Adversarial Networks paper

Official implementation of the ICML2021 paper "Elastic Graph Neural Networks"