pillow-dat 0.1.4

PIL × DAT - Pillow extension for AI-based image upscaling.

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}
}