![](https://user-images.githubusercontent.com/8450091/205468402-4240ad49-1449-4882-96a7-e27d9ec446b7.png)
Anime Style |
Interactive Colorization |
![](https://user-images.githubusercontent.com/8450091/205465788-86902f1a-0953-4f3b-8c7a-4d4f2dc25713.jpg) |
![](https://user-images.githubusercontent.com/8450091/205465819-58834a02-a437-4b47-93c8-ff3f3fc0672b.png) |
Portrait Mode |
Grayscale Background |
![](https://user-images.githubusercontent.com/8450091/205465880-b9549d26-aaf7-46dc-967f-d79148ac1f03.jpg) |
![](https://user-images.githubusercontent.com/8450091/205465881-fb1a61c7-734c-4c7d-b11c-edb14bf2deaa.jpg) |
Super-Resolution |
![](https://user-images.githubusercontent.com/8450091/205465965-6dc7cb45-d69d-4cd5-a7b0-f14f824a1227.jpg) |
White Balance Correction |
![](https://user-images.githubusercontent.com/8450091/205467129-64a3fad4-c4c6-4578-ba07-16e79dd94bd3.jpg) |
Instagram Filters |
Bezier Curves and Selective Editing |
![](https://user-images.githubusercontent.com/8450091/205467832-fd167e86-6b26-4d61-9fc8-bfd3ae5851cf.png) |
![](https://user-images.githubusercontent.com/8450091/205506806-c99a64d5-c34b-470d-adbf-11de2f6790a5.png) |
Spot Removal |
Exposure Adjustment |
![](https://user-images.githubusercontent.com/8450091/205468098-5fadd963-6c4e-4b1c-b430-4e989dab6fff.jpeg) |
![](https://user-images.githubusercontent.com/8450091/205470366-d85543ce-8d4a-4f02-926d-df1df0c4c07f.png) |
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
- 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
- White Balance Correction
- Background Removal
- Human Segmentation
- Grayscale Background
- Portrait Mode Background Blur
- Interactive Colorization
- Super-Resolution
- Anime Style
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