Skip to content

Multi-Memory Convolutional Neural Network for Video Super-Resolution

License

Notifications You must be signed in to change notification settings

psychopa4/MMCNN

Repository files navigation

This work is based on Tao et al

Datasets

We have collected 522 video sequences for training and 20 sequences for evaluation (mainly from documentaries), and in consider of copyright, the datasets should only be used for study.

The datasets can be downloaded from Google Drive, train and evaluation.

For reserchers who cannot get access to Google, you may visit Baiduyun, train and evaluation.

Unzip the training dataset to ./data/train/ and evaluation dataset to ./data/eval/ .

We only provide the ground truth images and the corresponding 4x downsampled LR images by Bicubic, and you may use PIL or Matlab to generate 2x or 3x downsampled LR images.

Environment

  • Python (Tested on 3.6)
  • Tensorflow >= 1.3.0

Training

  • python main.py to train model MMCNN-M10.

Testing

It should be easy to use 'testvideo()' or 'testvideos()' functions.

Citation

If you find our code or datasets helpful, please consider citing our work.

@ARTICLE{wang2018mmcnn,
        author = {Wang, Zhongyuan and Yi, Peng and Jiang, Kui and Jiang, Junjun and Han, Zhen and Lu, Tao and Ma, Jiayi},
        journal={IEEE Transactions on Image Processing},
        title = {Multi-Memory Convolutional Neural Network for Video Super-Resolution},
        year={2018},
    }

Contact

If you have questions or suggestions, please send email to yipeng@whu.edu.cn.

Visual Results

We show the visual results under 4x upscaling. This frame is from Videoset4 dataset.

Image text

This frame is from Myanmar test dataset.

Image text

About

Multi-Memory Convolutional Neural Network for Video Super-Resolution

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages