/NUl-Image-Enhancer

An algorithm based on decision trees and deep learning models to enhance images

Primary LanguagePythonGNU General Public License v2.0GPL-2.0

The Non-Ultimate Image Enhancer 🐌

An algorithm based on decision trees and deep learning models to enhance images.
The algorithm is simple, it does not always provide good results on already great images.
The default maximum runtime is 60 seconds / image. The algorithm can provide good results in 10 seconds on some images.
The algorithm tries multiple transformations on contrast / saturation / lighting etc., it evaluates the changes and keeps the best ones.

➡️ More details on the algorithm: How I Built an AI Image Enhancer

Results

Enhanced ✔️

Enhanced ✔️

Enhanced ❓

Requirements

  • A GPU with > 2GB VRAM (very slow on CPU)
  • CLIP
  • Pytorch

Usage

python3 enhance_image.py examples/1.png --maxruntime=10

or

Models

⭐ If you like it! Thanks!