kidanger / DMPHN-cvpr19-master

Pytorch Implementation of CVPR19 "Deep Stacked Multi-patch Hierarchical Network for Image Deblurring"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Deep Stacked Multi-patch Hierarchical Network for Image Deblurring

Pytorch Implementation of CVPR19 "Deep Stacked Multi-patch Hierarchical Network for Image Deblurring"

Pipeline of DMPHN

Please download GoPro dataset into './datas'.
https://drive.google.com/file/d/1H0PIXvJH4c40pk7ou6nAwoxuR4Qh_Sa2/view

GoPro Pretrained models are stored in './checkpoints'.

Requires.

pytorch-0.4.1
numpy
scipy

For model training, run following commands.

python xxx.py -b 6

For model testing, copy test samples into './test_samples', then run following commands.

python xxx_test.py

Citation

If you think this work is useful for your research, please cite the following paper.

@InProceedings{Zhang_2019_CVPR,
author = {Zhang, Hongguang and Dai, Yuchao and Li, Hongdong and Koniusz, Piotr},
title = {Deep Stacked Hierarchical Multi-Patch Network for Image Deblurring},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2019}
}

About

Pytorch Implementation of CVPR19 "Deep Stacked Multi-patch Hierarchical Network for Image Deblurring"


Languages

Language:Python 100.0%