Model Type: Custom PyTorch Model
Task: Watermark extraction from watermarked images
Source: Derived from facebookresearch/stable_signature
License: CC-BY-NC 4.0 (Attribution + Non-Commercial)
Framework: PyTorch
Weights: Ported from the original TorchScript decoder
Architecture:
- HiddenDecoder(num_blocks, num_bits, channels, redundancy)
- MsgExtractor(hidden_decoder, in_features, out_features)
Status: Research-only, non-commercial

Load the model

python

from modeling_msg_extractor import MsgExtractor
import torch

model = MsgExtractor.from_pretrained("ESmike/StableSignatureDecoder")
model.eval()

img = torch.randn(1, 3, 256, 256)   # example input
bits = model(img)
print(bits.shape)

Citation

If you use this model, you must cite the original Stable Signature paper:

bibtex

@inproceedings{Fernandez2023StableSignature,
  title={The Stable Signature Rooting Watermarks in Latent Diffusion Models},
  author={Fernandez, Pierre and Chappelier, Vivien and Nguyen-Hong, Son},
  year={2023},
  institution={Meta AI},
  note={Original implementation at https://github.com/facebookresearch/stable_signature}
}