SRGAN-Keras

KO | EN

gitlite — search

#javascript #typescript #golang #machine-learning #docker #macos #windows #awesome #cli #kubernetes #vue #security #mcp #openai

(★ 82)

Implementation of SRGAN in Keras. Try at: www.fixmyphoto.ai

File Explorer

Download Latest Version (.zip)

- - .gitignore
- .gitignore
- buket.jpg
- sample.jpg
- .gitignore
- architecture.PNG
- code_setup.PNG
- loss_equations.PNG
- main_sample.PNG
- main_samples_with_psnr.png
- __init__.py
- srgan.py
- util.py
.gitignore
Example_Usage.ipynb
LICENSE
README.md
train.py

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# SRGAN-Keras
Keras implementation of ["Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network"](https://arxiv.org/abs/1609.04802)
<img src='images/main_samples_with_psnr.png' width="100%" />

## 1. Architecture
The generator creates a high-resolution (HR) image (4x upscaled) from a corresponding low-resolution (LR) image. The discriminator distinguishes the generated (fake) HR images from the original HR images.

### 1.1. Generator & Discriminator
<img src='images/architecture.PNG' width="80%" />

[Figure 4 from paper](https://arxiv.org/abs/1609.04802): Architecture of Generator and Discriminator Network with corresponding kernel size (k), number of feature maps
(n) and stride (s) indicated for each convolutional layer.

### 1.2. Overview of input / outputs
<img src='images/code_setup.PNG' width="80%" />

**Code Overview**: Overview of the three networks; generator, discriminator, and VGG19. Generator create SR image from LR, discriminator predicts whether it's a SR or original HR, and VGG19 extracts features from generated SR and original HR images.

## 2. Content & Adversarial Loss
<img src='images/loss_equations.PNG' width="80%" />

**Losses Overview**: The perceptual loss is a combination of content loss (based on VGG19 features) and adversarial loss. Equations are taken directly from ["original paper"](https://arxiv.org/abs/1609.04802).

## 3. Using this repository

### 3.1. Training
A command-line interface can be found in `train.py`. To train run e.g.:
```
python train.py \
    --train <TRAINING_IMAGES_PATH> \
    --validation <VALIDATION_IMAGES_PATH> \
    --scale 4 \
    --test_path images/samples_4X \
    --stage all
```

### 3.2. Testing
Check the example_usage notebook: [example_usage.ipynb](./Example_Usage.ipynb)

SRGAN-Keras

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

SRGAN-Keras

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

Download README