super-image 0.1.0

State-of-the-art image super resolution models for PyTorch.

super-image

State-of-the-art image super resolution models for PyTorch.

Requirements

super-image requires Python 3.6 or above.

Installation

With pip:

pip install super-image

With pipx:

python3.6 -m pip install --user pipx

pipx install --python python3.6 super-image

Quick Start

from super_image import EdsrModel, ImageLoader
from PIL import Image
import requests

url = 'https://paperswithcode.com/media/datasets/Set5-0000002728-07a9793f_zA3bDjj.jpg'
image = Image.open(requests.get(url, stream=True).raw)

model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
inputs = ImageLoader.load_image(image)
preds = model(inputs)

ImageLoader.save_image(preds, './scaled_2x.png')
ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png')