BaMBNet-Pytorch

Paper: https://arxiv.org/abs/2105.14766 (arXiv)

The code is for the work:

@article{liang2021BaMBNet,
  title={BaMBNet: A Blur-aware Multi-branch Network for Defocus Deblurring},
  author={Pengwei Liang, Junjun Jiang, Xianming Liu, and Jiayi Ma},
  journal={submitted to IEEE Transactions on Computational Imaging},
  volume={},
  number={},
  pages={},
  year={2021},
}

Requirements

pytorch == 1.7.1
kornia == 0.4.1
opencv == 4.4.0

Dataset

Please refer to the official repo at Defocus deblurring using dual-pixel data.

Train

• Step 1: train COC network to estimate the blur amounts of DP data.

  python blur_train.py -opt option/train/COC_Dataset_Train.yaml

• Step 2: prepare the COC maps for deblurring training.

  python blur_test.py -opt option/test/COC_Dataset_Test.yaml

• Step 3: train the deblurred network.

  python train.py -opt option/train/Deblur_Dataset_Trained.yaml

Test

python test.py -opt option/test/Deblur_Dataset_Test.yaml

License

This project is under the CC-BY-NC 4.0 license. See LICENSE for details.