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 get the watermark from a image? #1

Closed jinganglang567 closed 6 months ago

jinganglang567 commented 6 months ago

Sorry to bother you doctor,if i want to visualize the watermark before generate the image and after generate the image

bsmhmmlf commented 6 months ago

In Gaussian Shading, the watermark information is directly mapped to latent representations. If you want to visualize the watermark, you can inspect the variables init_latents_w and reversed_latents_w in the code for visualization.

Sorry to bother you doctor,if i want to visualize the watermark before generate the image and after generate the image

jinganglang567 commented 6 months ago

thank your answer,doctor