- Python 3 (Anaconda is recommended)
- skimage
- imageio
- Pytorch (Pytorch version >= 2.0.0 is recommended)
- tqdm
- pandas
- cv2 (pip install opencv-python)
Please run:
bash scripts/train.sh
Please run:
bash scripts/test.sh
If you find this work useful for your research, please consider citing it.
@article{ma2023uncertainty,
title={Uncertainty-aware image inpainting with adaptive feedback network},
author={Ma, Xin and Zhou, Xiaoqiang and Huang, Huaibo and Jia, Gengyun and Wang, Yaohui and Chen, Xinyuan and Chen, Cunjian},
journal={Expert Systems with Applications},
pages={121148},
year={2023},
publisher={Elsevier}
}