BasicVSR_PlusPlus (CVPR 2022)
[Paper] [Project Page] [Code]
This is the official repository for BasicVSR++. Please feel free to raise issue related to BasicVSR++! If you are also interested in RealBasicVSR, which is also accepted to CVPR 2022, please don't hesitate to star!
Authors: Kelvin C.K. Chan, Shangchen Zhou, Xiangyu Xu, Chen Change Loy, Nanyang Technological University
Acknowedgement: Our work is built upon MMEditing. Please follow and star this repository and MMEditing!
News
- 2 Dec 2021: Colab demo released
- 18 Apr 2022: Code released. Also merged into MMEditing
TODO
- Add data processing scripts
-
Add checkpoints for deblur and denoise -
Add configs for deblur and denoise -
Add Colab demo
Pre-trained Weights
You can find the pre-trained weights for deblurring and denoising in this link. For super-resolution and compressed video enhancement, please refer to MMEditing.
Installation
- Install PyTorch
pip install openmim
mim install mmcv-full
git clone https://github.com/ckkelvinchan/BasicVSR_PlusPlus.git
cd BasicVSR_PlusPlus
pip install -v -e .
Inference a Video
- Download pre-trained weights
python demo/restoration_video_demo.py ${CONFIG} ${CHKPT} ${IN_PATH} ${OUT_PATH}
For example, you can download the VSR checkpoint here to chkpts/basicvsr_plusplus_reds4.pth
, then run
python demo/restoration_video_demo.py configs/basicvsr_plusplus_reds4.py chkpts/basicvsr_plusplus_reds4.pth data/demo_000 results/demo_000
You can also replace ${IN_PATH} ${OUT_PATH}
by your video path (e.g., xxx/yyy.mp4
) to input/output videos.
Training Models
- Put the dataset in the designated locations specified in the configuration file.
sh tools/dist_train.sh ${CONFIG} ${NGPUS}
Data Preprocessing
To be added...
Related Work
Our BasicVSR series:
- BasicVSR: The Search for Essential Components in Video Super-Resolution and Beyond, CVPR 2021
- Investigating Tradeoffs in Real-World Video Super-Resolution, CVPR 2022
More about deformable alignment:
Citations
@inproceedings{chan2022basicvsrpp,
author = {Chan, Kelvin C.K. and Zhou, Shangchen and Xu, Xiangyu and Loy, Chen Change},
title = {{BasicVSR++}: Improving video super-resolution with enhanced propagation and alignment},
booktitle = {IEEE Conference on Computer Vision and Pattern Recognition},
year = {2022}
}
@article{chan2022generalization,
title={On the Generalization of {BasicVSR++} to Video Deblurring and Denoising},
author={Chan, Kelvin CK and Zhou, Shangchen and Xu, Xiangyu and Loy, Chen Change},
journal={arXiv preprint arXiv:2204.05308},
year={2022}
}