Skip to content

Latest commit

 

History

History
172 lines (114 loc) · 5.75 KB

README.md

File metadata and controls

172 lines (114 loc) · 5.75 KB

EnhanceIMG

[TOC]

此代码库用于图像增强算法的探索,主要包括:低光增强、图像修复、超分辨率重建 ……

目录结构

.
|--- asserts/
|--- awegan/      # GAN相关算法
     |--- datasets/
     |--- models/         # CycleGAN/Pix2Pix/SelfGAN
     |--- options/
     |--- util/
     |--- __init__.py
     |--- train.py
     |--- ...
|--- colorspace/  # 色彩空间转换
|--- edges/       # 边缘检测算法
|--- filters/     # 各种滤波器
|--- histeq/      # 直方图均衡算法
|--- noises/      # 噪声
|--- priors/      # 自然图像先验信息
     |--- __init__.py
     |--- denoising.py
     |--- inpainting.py
     |--- networks.py     # ResNet/SkipNet/UNet
     |--- restoration.py
     |--- ...
|--- retinex/     # Retinex系列算法
     |--- __init__.py
     |--- enhancer.py
     |--- retinex_net.py  # RetinexNet
     |--- ...
|--- utils/       # 一些方法
|--- .gitignore
|--- demo.py
|--- LICENSE
|--- Madison.png
|--- README.md    # 说明文档
|--- requirements.txt     # 依赖文件

简单示例

添加噪声

噪声(原图|椒盐噪声|高斯噪声)

noises

各种滤波器

滤波器(椒盐噪声|均值滤波|中值滤波)

filters1

滤波器(高斯噪声|高斯滤波|双边滤波|联合双边滤波)

filters2

滤波器(高斯噪声|引导滤波)

filters3

边缘检测

检测算子(灰度图|Laplacian|Sobel|Scharr)

opt-edge-detection-2

检测算子(灰度图|LoG|DoG|Gabor)

opt-edge-detection-3

其他算法(灰度图|结构森林|HED|HED-feats-5)

hn-edge-detection

hed-fs1-fs5

传统增强算法

直方图均衡(原图|HE|AHE|CLAHE)

hist-equal

Gamma 校正(原图|Gamma|Gamma+MSS)

adjust-gamma

Retinex(原图|MSRCR|AMSRCR|MSRCP)

retinex

Retinex 增强(原图|AttnMSR|AttnMSR+MSS)(Mine)

enlighten

自然图像先验

降噪(噪声图|降噪1|降噪2)

prior-denoising

神经网络

RetinexNet(原图|RetinexNet)

retinexnet

生成对抗网络

Pix2Pix

(边缘 <=> 图像)

pix2pix-facades

(低光 <=> 正常)

pix2pix

pix2pix4

CycleGAN

(夏天 <=> 冬天)

summer2winter

(低光 <=> 正常)

cyclegan4

参考资料

TODO

  • AttnMSR 图像增强算法(Mine)
  • RetinexNet 低光增强模型
  • ResNet / SkipNet / UNet
  • Deep Image Prior(自然图像先验信息)
  • Pix2Pix 模型用于图像增强
  • CycleGan 模型用于图像增强
  • SelfGAN 图像增强模型(Mine,完善中)

欢迎交流

  • 微信号:YcZhouZy

  • 企鹅号:793071559