/PhotoLab

AI-Powered Photo Editor (Python, PyQt6, PyTorch)

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

PhotoLab

Anime Style Interactive Colorization
Portrait Mode Grayscale Background
Super-Resolution
White Balance Correction
Instagram Filters Bezier Curves and Selective Editing
Spot Removal Exposure Adjustment

Quick Start

Download the pretrained models by running the included download script:

foo:bar$ python download_models.py

Start the editor by running:

foo:bar$ python src/main.py

Features

Basic Tools

  • Zoom and Pan
  • Color Picker
  • Rectangle Select
  • Path Select
  • Crop
  • Paint and Erase
  • Exposure and Color Adjustment
  • Histogram Viewer
  • Curve Editor
  • Spot Removal Tool
  • Blur Tool
  • Rotate Left/Right
  • Horizontal/Vertical Stack
  • Horizontal/Vertical Flip
  • Panorama Image Stitching
  • Instagram Filters

AI Tools

  • White Balance Correction
  • Background Removal
  • Human Segmentation
  • Grayscale Background
  • Portrait Mode Background Blur
  • Interactive Colorization
  • Super-Resolution
  • Anime Style

Example Results

Before After

Notes on PyTorch CUDA Support

This project has been testing with torch==1.11.0+cu113 torchvision-0.12.0+cu113

foo:bar$ pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html

foo:bar$ python
Python 3.10.8 (tags/v3.10.8:aaaf517, Oct 11 2022, 16:50:30) [MSC v.1933 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.version.cuda
'11.3'
>>> torch.cuda.is_available()
True