/U-2Pnet

GNU General Public License v3.0GPL-3.0

U2PNet: An Unsupervised Underwater Image-Restoration Network Using Polarization

Dependencies

opencv-contrib-python==4.6.0.66

opencv-python==4.6.0.66

Python == 3.6.10

scikit-image==0.17.2

scikit-video==1.1.11

torch==1.1.0

torchvision==0.3.0

Datasets

Real Underwater Polarization Data are released.

We use a polarization camera with micro polarization filters array (LUCID, PHX050S), whose spatial resolution is 2448 × 2048. The acquired mosaicked polarization image can be demosaicked using PCDC or Demosaicking-DoFP-images-using-Newton-polynomial-interpolation-and-polarization-difference-model.

Images

Underwater Enhancement

Dehazing

Videos