Views
No views yet
1import torch
2from PIL import Image
3import numpy as np
4from RealESRGAN import RealESRGAN
5
6device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
7
8model = RealESRGAN(device, scale=4)
9model.load_weights('weights/RealESRGAN_x4.pth', download=True)
10
11path_to_image = 'inputs/lr_image.png'
12image = Image.open(path_to_image).convert('RGB')
13
14sr_image = model.predict(image)
15
16sr_image.save('results/sr_image.png')