ESRGAN-TensorFlow

KO | EN

gitlite — search

#javascript #python #react #typescript #machine-learning #rust #deep-learning #docker #windows #awesome #cli #kubernetes #awesome-list #mcp

ESRGAN-TensorFlow

(★ 18)

Implementation of ESRGAN using TensoFlow

ESRGAN-TensorFlow 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# ESRGAN-TensorFlow
A implementation of ESRGAN using Tensorflow.

original paper: https://arxiv.org/abs/1809.00219

original implementation: https://github.com/xinntao/ESRGAN

see also [my SRGAN implementation.](https://github.com/itsuki8914/SRGAN-TensorFlow)

## Usage

1. Download vgg19.npy from [here.](https://github.com/machrisaa/tensorflow-vgg)

Put vgg19.npy in the folder where convert.py and vgg.py are located.
  
  like this
```
...
│
├──convert.py
├──vgg.py
├──vgg19.npy
...
```

run convert.py.
  
```
python convert.py
```

after running, A vgg model dedicated to tensorflow will be output.
 
 ```
 ...
│
├── convert.py
├── vgg.py
├── vgg19.npy
├── modelvgg
│     ├── checkpoint
│     ├── model.ckpt-0
│     ├── model.ckpt-0
│     └── model.ckpt-0
...
```

2. Download dataset from [DIV2K dataset.](https://data.vision.ee.ethz.ch/cvl/DIV2K/)

download the following four.

* Train Data Track 1 bicubic downscaling x4 (LR images)
* Train Data (HR images)
* Validation Data Track 1 bicubic downscaling x4 (LR images)
* Validation Data (HR images)

(If your PC has enough RAM, I recommend also The flickr2K data set proposed in [EDSR.](https://github.com/LimBee/NTIRE2017))

Put the downloaded dataset in the train_lr, train_hr, val_lr and val_hr folders.
  like this
```
...
│
├── train_lr
│     ├── 0001x4.png
│     ├── 0002x4.png
│     ├── ...
│     └── 0800x4.png
├── train_hr
│     ├── 0001.png
│     ├── 0002.png
│     ├── ...
│     └── 0800.png 
├── val_lr
│     ├── 0801x4.png
│     ├── 0802x4.png
│     ├── ...
│     └── 0900x4.png
├── val_hr
│     ├── 0801.png
│     ├── 0802.png
│     ├── ...
│     └── 0900.png 
├── main.py
├── model.py
...
```

3. Train ESRGAN.

```
python main.py
```

4. After training, inference can be performed.

Put the images you want to evaluate into a folder and run the following.

example:
 ```
 ...
│
├── main.py
├── pred.py
├── modelGAN
│     ├── checkpoint
│     ├── model.ckpt-100000
│     ├── model.ckpt-100000
│     └── model.ckpt-100000
├── test
│     ├── yourpic1.png
│     ├── favpic.jpg
│     ...
│     └── smallpic.png
...
```

```
python pred.py test
```

## Result examples

LR and HR images from [jbhuang0604/SelfExSR](https://github.com/jbhuang0604/SelfExSR).

leftmost: bicubic upsampling

Second from left: SRGAN([my implementation](https://github.com/itsuki8914/SRGAN-TensorFlow))

Third from left: ESRGAN(this implementation)

rightmost: original HR image

ESRGAN-TensorFlow

ESRGAN-TensorFlow 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

ESRGAN-TensorFlow

ESRGAN-TensorFlow 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README