- Implementação oficial do artigo da Meta, "Watermark Anything with Localized Messages"
- Fornece uma forma de inserir marcas d'água localizadas em imagens
- Permite inserir várias marcas d'água
- Requisitos: testado nas versões Python 3.10.14, PyTorch 2.5.1, CUDA 12.4 e Torchvision 0.20.1.
- Inference
- Em
notebooks/inference.ipynb, é possível ver os scripts de inferência e as visualizações
- Explica como inserir, detectar e decodificar marcas d'água em imagens
- É possível ajustar o equilíbrio entre imperceptibility/robustness (imperceptibilidade/robustez) da marca d'água ajustando o argumento
wam.scaling_w
- Marca d'água única
- Fornece um exemplo de inserção de uma mensagem de 32 bits em uma imagem
- Gera uma máscara aleatória para inserir a marca d'água apenas em parte da imagem
- Múltiplas marcas d'água
- Fornece um exemplo de inserção de várias mensagens de 32 bits em uma imagem
- Detecta marcas d'água por meio de clustering usando o algoritmo DBSCAN
- Treinamento
- Pré-treinamento: explica o método de pré-treinamento para robustez.
- Fine-tuning: explica como fazer o fine-tuning do modelo para suportar múltiplas marcas d'água e garantir a imperceptibilidade.
- Licença
- O modelo é fornecido sob a licença CC-BY-NC.
- Referências
- Relacionado aos projetos AudioSeal e Segment Anything.
1 comentários
Comentários no Hacker News