Saint-lsy/Polyp-Gen-sd2-inpainting

Polyp-Gen: Realistic and Diverse Polyp Image Generation for Endoscopic Dataset Expansion

Polyp-Gen is a text-guided full-automatic diffusion-based endoscopic image generation framework for realistic and diverse polyp image generation for endoscopic dataset expansion, as presented in Polyp-Gen: Realistic and Diverse Polyp Image Generation for Endoscopic Dataset Expansion. You can use our model for polyp generation.

Code is available here.

This model card is based on stable-diffusion-2-inpainting model, available here.

Citation

If you find this work helpful, please consider to star🌟 this repo and cite the following paper:

@article{liu2025polyp,
  title={Polyp-Gen: Realistic and Diverse Polyp Image Generation for Endoscopic Dataset Expansion},
  author={Liu, Shengyuan and Chen, Zhen and Yang, Qiushi and Yu, Weihao and Dong, Di and Hu, Jiancong and Yuan, Yixuan},
  journal={arXiv preprint arXiv:2501.16679},
  year={2025}
}