PIL × DAT - Pillow extension for AI-based image upscaling.
Project description
PIL × DAT - Pillow extension for AI-based image upscaling.
Installation
For PyPI:
pip install pillow-dat
Get started
-
- Download
DAT_light_x2.pth
model weight from Google Drive. We highly recommend utilizingDAT_light_x*.pth
models due to their lightweight design and exceptional speed.
- Download
-
- 🎉 Then you are all set to upscale your images:
from PIL.Image import open
from PIL_DAT.Image import upscale
image = open("./lumine.png")
image = upscale(lumine_image, "./DAT_light_x2.pth", 2)
image.show()
Contribution
Please install Miniconda.
Please install VSCode extensions:
- Black Formatter
- isort
- Python
- Pylance
To create or update the pillow-dat
Python environment:
conda env create --file environment.yml
conda env update --file environment.yml --prune
To install dependencies:
poetry install
To run unit tests:
pytest
Acknowledgement
This library is founded upon the pioneering research paper, "Dual Aggregation Transformer for Image Super-Resolution".
@inproceedings{chen2023dual,
title={Dual Aggregation Transformer for Image Super-Resolution},
author={Chen, Zheng and Zhang, Yulun and Gu, Jinjin and Kong, Linghe and Yang, Xiaokang and Yu, Fisher},
booktitle={ICCV},
year={2023}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pillow_dat-0.1.3.tar.gz
(16.6 kB
view hashes)
Built Distribution
pillow_dat-0.1.3-py3-none-any.whl
(19.2 kB
view hashes)
Close
Hashes for pillow_dat-0.1.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c8d0d7ad456da27e6a15d7f65733aa693ef5a5db4cf10a4da7a93cd3507a6b0b |
|
MD5 | 800c8483d694ae718a28524e0383100b |
|
BLAKE2b-256 | e6663aa5571d849eb2de95c49ccbb5df35de134ed46aef934f5aeecf14341a17 |