bsmhmmlf / Gaussian-Shading

[CVPR 2024] Gaussian Shading: Provable Performance-Lossless Image Watermarking for Diffusion Models
MIT License
63 stars 3 forks source link

How to separate watermark generation and detection processes? #9

Open XuandongZhao opened 2 months ago

XuandongZhao commented 2 months ago

Hi,

Thank you for open-sourcing this code! I understand that the current implementation integrates watermark generation and detection within the same for loop.

Would it be possible to separate the watermark generation and detection processes, such as by storing the watermark key and loading it from files? This approach would better reflect real-world scenarios.

Thank you!