whz1861 / deblur-gan

Keras implementation of "DeblurGAN: Blind Motion Deblurring Using Conditional Adversarial Networks"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

What is this repo ?

This repository is a Keras implementation of Deblur GAN. You can find a tutorial on how it works on Medium.

Installation

virtualenv venv -p python3
. venv/bin/activate
pip install -r requirements.txt

Dataset

Get the GOPRO dataset, and extract it in the deblur-gan directory. The directory name should be GOPRO_Large.

Use:

python organize_gopro_dataset.py --dir_in=GOPRO_Large --dir_out=images

Training

python train.py --n_images=512 --batch_size=16

Use python train.py --help for all options

Testing

python test.py

Use python test.py --help for all options

Deblur your own image

python deblur_image.py --image_path=path/to/image

About

Keras implementation of "DeblurGAN: Blind Motion Deblurring Using Conditional Adversarial Networks"


Languages

Language:Python 100.0%