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
from PIL.Image import open
from PIL_DAT.Image import upscale
lumine_image = open("./lumine.png")
lumine_image = upscale(lumine_image, 2)
lumine_image.show()
Remark: We strongly advocate for the utilization of DAT light
models owing to their streamlined design and outstanding speed performance. However, should you opt for alternative models, please note that *.pth
model weights can be accessed via Google Drive.
Example
Input (lumine.png) | Output | Bicubic |
---|---|---|
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
Built Distribution
Hashes for pillow_dat-0.1.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | dc72f22691ca406151b76691ee025d41cc022b518bec08553ed761d75b14aee2 |
|
MD5 | 955bd8e76869809462b2266a275f91ec |
|
BLAKE2b-256 | cd4d59b28129d11667a8db2d7da7a15910814be2ac5b04ded522b8bb159ff619 |