blackprotoss / GSDM

Text Image Inpainting via Global Structure-Guided Diffusion Models (Accepted by AAAI-24)
MIT License
52 stars 5 forks source link

Text Image Inpainting via Global Structure-Guided Diffusion Models (Accepted by AAAI-24)

Shipeng Zhu, Pengfei Fang, Chenjie Zhu, Zuoyan Zhao, Qiang Xu, Hui Xue

Paper: (arXiv 2401.14832), (AAAI-24)

This repository offers the official Pytorch code for this paper. If you have any questions, feel free to contact Shipeng Zhu (shipengzhu@seu.edu.cn) or Chenjie Zhu (chenjiezhu@seu.edu.cn).

Environment Setup

python pytorch cuda

Inference phase

python inference.py --config xx --input_dir input --output_dir output --save_sp False

Datasets and Pre-trained Checkpoints

Training phase

Step 1: Training SPM

python train_spm.py

Step 2: Training RM

python train_rm.py

Citation

@inproceedings{zhu2024gsdm,
  title={Text image inpainting via global structure-guided diffusion models},
  author={Zhu, Shipeng and Fang, Pengfei and Zhu, Chenjie and Zhao, Zuoyan and Xu, Qiang and Xue, Hui},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  volume={38},
  number={7},
  pages={7775-7783},
  year={2024}
}