Output images ?

[fudanvi]

Output images ?

[JingyeChen]

You could modify the function "demo" in super_resolution.py to achieve this

[kanika02]

Can you give the code for the following?

[steven8274]

You can add this line:
torchvision.utils.save_image(images_sr[:, :3, :, :], os.path.join(self.args.demo_dir, 'sr_'+im_name), padding=0)
just after 'images_sr' generated at:
for im_name in tqdm(os.listdir(self.args.demo_dir)): images_lr = transform_(os.path.join(self.args.demo_dir, im_name)) images_lr = images_lr.to(self.device) sr_beigin = time.time() images_sr = model(images_lr)