pillow-dat 0.1.0

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

For Conda:

conda install -c conda-forge pillow-dat

Get started

• 1. Download DAT_light_x2.pth model weight from Google Drive. We highly recommend utilizing DAT_light_x*.pth models due to their lightweight design and exceptional speed.
• 2. 🎉 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 install dependencies on a given pillow-dat Python environnement:

conda env create --file environment.yml

To update this environment:

conda env update --file environment.yml --prune

To run unit tests:

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