p-ranav / photolab

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

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

photolab's Issues

All functions does not work after running

I launched the program, successfully installed all the requirements, but nothing inside works from the panel on the left, and if I insert a picture via cntrl v, the panel disappears

QToolPortraitMode

can you explain why background from the original image was removed here when it wasn't used throughout the code

Cannot run

How can I run it locally. I installed the required packages in a virtual environment and cant run it. Can you tell me a way to run it without any errors regarding the dependencies

Cannot run

I have no error I imported all modules but the PhotoLab is not working. I see the app but there is no option to add image. Please help.

Regarding Collobaration

Hi, I'm Sameer . I'm quite moved by your PhotoLab project and I want to collaborate with your project . My Project is named Photolab (Similar to your project name : ) ) . It is a set of tools in pure image processing . Please do review my project and it would be a great bliss if we could merge our projects . The Project link is https://github.com/bvks2020sameer/Photolab

No window to upload photo to photolab

After running main.py, photolab opens up but there is no option to upload my image to edit.

The byte size of image edited is large

When I open image and save as. The byte size of image is bigger than original image. How can i keep the byte size