[ECCV2024] Robust-Wide: Robust Watermarking Against Instruction-Driven Image Editing

April 8, 2025 · View on GitHub

Official implementation of Robust-Wide: Robust Watermarking Against Instruction-Driven Image Editing.

Train

  1. Download the data and put them into the data dir ./data.

  2. Configure the train script and then run it.

bash train.sh

Inference

  1. Put your original image in ./examples.

  2. Download the checkpoints and put them in ./checkpoints.

  3. Configure the inference script and then run it.

bash inference.sh

Evaluate

To reproduce the results presented in our paper, download the data and place it in the ./eval_data directory. Then, run:

bash evaluate.sh

Acknowledgements

This code builds on the code from the diffusers library.

Cite

If you find this repository useful, please consider giving a star ⭐ and please cite as:

@inproceedings{hu2025robust,
  title={Robust-wide: Robust watermarking against instruction-driven image editing},
  author={Hu, Runyi and Zhang, Jie and Xu, Ting and Li, Jiwei and Zhang, Tianwei},
  booktitle={European Conference on Computer Vision},
  pages={20--37},
  year={2025},
  organization={Springer}
}