maxin-cn / U2AFN

[ESWA 2023] Uncertainty-aware image inpainting with adaptive feedback network.

Uncertainty-Aware Image Inpainting with Adaptive Feedback Network

Requirements

• Python 3 (Anaconda is recommended)
• skimage
• imageio
• Pytorch (Pytorch version >= 2.0.0 is recommended)
• tqdm
• pandas
• cv2 (pip install opencv-python)

Train

Please run:

bash scripts/train.sh

Test

Please run:

bash scripts/test.sh

Citation

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}
}