Dominic-ZZ / Dehaze-YUVRDNet

A single-image dehazing algorithm based on GAN network for color restoration in YUV color space is proposed

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Single Image Dehazing Algorithm Based on YUV Color Space

A single-image dehazing algorithm based on GAN network for color restoration in YUV color space is proposed(YUVRecolorDehazeNet, YUVRDNet). Inspired by the GAN image coloring task, the algorithm converts the hazy image to YUV color space, weakens the effect of haze in the Y channel, and performs color restoration on the image in the UV channel to achieve dehazing. Meanwhile, skip connections are used to fuse low-level and high-level features for more precise dehazing.

Pre-trained model:

  • Baidu Cloud: url, Extraction code: yic3
  • Google Drive: url

About

A single-image dehazing algorithm based on GAN network for color restoration in YUV color space is proposed

License:Mozilla Public License 2.0


Languages

Language:Python 100.0%