File size: 2,475 Bytes
a010745
 
 
 
 
 
df530f5
 
 
 
a010745
 
d6aec7e
b3c2db5
5da3e0f
 
b3c2db5
 
 
 
d6aec7e
b3c2db5
df530f5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b3c2db5
 
 
 
 
 
 
df530f5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
---
tags:
- Keras
license: mit
metrics:
- PSNR
datasets:
- eugenesiow/Div2k
library_name: keras
pipeline_tag: image-to-image
---

Here is a fully trained model of EDSR (Enhanced Deep Residual Networks for Single Image Super-Resolution) model. This model surpassed the performance of the current available SOTA models.

Spaces link - https://huggingface.co/spaces/keras-io/EDSR

Paper Link - https://arxiv.org/pdf/1707.02921

Keras Example link - https://keras.io/examples/vision/edsr/

It was trained for 500 epochs with 200 steps each.



# Enhanced Deep Residual Networks for Single Image Super-Resolution

## Introduction

This repository contains a trained model based on the Enhanced Deep Residual Networks for Single Image Super-Resolution paper. The model was trained for 500 epochs with 200 steps each, resulting in a high-quality super-resolution model.

## Dataset Used

The model was trained on the DIV2K dataset, which is a newly proposed high-quality (2K resolution) image dataset for image restoration tasks. The DIV2K dataset consists of 800 training images, 100 validation images, and 100 test images.

## Architecture

The Enhanced Deep Residual Networks for Single Image Super-Resolution paper presents an enhanced deep super-resolution network (EDSR) and a new multi-scale deep super-resolution system (MDSR) that outperform current state-of-the-art SR methods. The EDSR model optimizes performance by analyzing and removing unnecessary modules to simplify the network architecture. The MDSR system is a multi-scale architecture that shares most of the parameters across different scales, using significantly fewer parameters compared with multiple single-scale models but showing comparable performance.

## Metrics

The model was evaluated using the PSNR (Peak Signal-to-Noise Ratio) metric, which measures the quality of the reconstructed image compared to the original image. The model achieved a PSNR of approximately 31, which is a high-quality result.

##  TODO:

Hack to make this work for any image size. Currently the model takes input of image size 150 x 150.
We pad the input image with transparant pixels so that it is a square image, which is a multiple of 150 x 150
Then we chop the image into multiple 150 x 150 sub images
Upscale it and stich it together.

The output image might look a bit off, because each sub-image dosent have data about other sub-images.
This approach assumes that the sub-image has enough data about its surroundings