From 0077286a9d9dac1f5c6fb3c32d37cbf590f657f7 Mon Sep 17 00:00:00 2001 From: Anindya Maiti Date: Mon, 25 Sep 2023 17:02:53 -0500 Subject: [PATCH] Update 'README.md' --- README.md | 25 +++++++++++++++---------- 1 file changed, 15 insertions(+), 10 deletions(-) diff --git a/README.md b/README.md index 4cb207b..d74ca4d 100644 --- a/README.md +++ b/README.md @@ -96,16 +96,21 @@ Most universal is [EDSR](https://huggingface.co/eugenesiow/edsr-base). Sample py 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) # scale 2, 3 and 4 models available -inputs = ImageLoader.load_image(image) -preds = model(inputs) - -ImageLoader.save_image(preds, './scaled_2x.png') # save the output 2x scaled image to `./scaled_2x.png` -ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png') # save an output comparing the super-image with a bicubic scaling +import os +import time + +modelx2 = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2) +modelx3 = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=3) +modelx4 = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=4) + +for filename in os.listdir("./Input"): + if filename.endswith('.png'): + print(filename) + image = Image.open("./Input/" + filename) + inputs = ImageLoader.load_image(image) + preds = modelx4(preds) + ImageLoader.save_image(preds, "./Output/upscaled-" + filename) + del image, inputs, preds ``` ## Read More: