StudioSR is a Pytorch library providing implementations of training and evaluation of super-resolution models. StudioSR aims to offer an identical playground for modern super-resolution models so that researchers can readily compare and analyze a new idea. (inspired by PyTorch-StudioGan)
pip install studiosrgit clone https://github.com/veritross/studiosr.git
cd studiosr
python3 -m pip install -e .Documentation along with a quick start guide can be found in the docs/ directory.
from studiosr.models import SwinIR
from studiosr.utils import imread, imwrite
model = SwinIR.from_pretrained(scale=4).eval()
image = imread("image.png")
upscaled = model.inference(image)
imwrite("upscaled.png", upscaled)StudioSR is an open-source library under the MIT license.