/BGShadowNet

Primary LanguagePython

Document Image Shadow Removal Guided by Color-Aware Background

Requirements

  • python 3.6.13
  • pytorch 1.10.0
  • numpy==1.19.2
  • opencv_python_headless==4.5.4.60

TRAIN

  1. You can obtain RDD dataset from here
  2. Modify the config.yaml to set your parameters
  3. You can generate the csv file you need by configuring, running utils/make_dataset.py
  4. Use extractBackground.py to construct the ground-truth background
  5. Training CBENet

python Train_CBENet.py ./configs/model=CBENet/config.yaml

  1. Training BGShadowNet

python Train_BGShadowNet.py ./configs/model=BGShadowNet/config.yaml

  1. Pretrained model

Download the pretrained model trained on RDD dataset.

TEST

python test.py

Cite

  @InProceedings{Zhang_2023_CVPR,
  author  = {Ling Zhang,Yinghao He,Qing Zhang,Zheng Liu,Xiaolong Zhang,Chunxia Xiao},
  title   = {Document Image Shadow Removal Guided by Color-Aware Background.},
  booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  month   = {June},
  year   = {2023}}