- Implementación oficial del artículo de Meta, "Watermark Anything with Localized Messages"
- Ofrece una forma de insertar marcas de agua localizadas en imágenes
- Permite insertar múltiples marcas de agua
- Requisitos: probado en las versiones Python 3.10.14, PyTorch 2.5.1, CUDA 12.4 y Torchvision 0.20.1.
- Inferencia
- En
notebooks/inference.ipynb se pueden revisar el script de inferencia y la visualización
- Explica cómo insertar una marca de agua en una imagen, detectarla y decodificarla
- Ajustando el argumento
wam.scaling_w se puede equilibrar la imperceptibilidad/robustez de la marca de agua
- Marca de agua única
- Incluye un ejemplo para insertar un mensaje de 32 bits en una imagen
- Genera una máscara aleatoria para insertar la marca de agua solo en una parte de la imagen
- Múltiples marcas de agua
- Incluye un ejemplo para insertar varios mensajes de 32 bits en una imagen
- Detecta marcas de agua mediante clustering usando el algoritmo DBSCAN
- Entrenamiento
- Preentrenamiento: explica el método de preentrenamiento para lograr robustez.
- Ajuste fino: explica cómo hacer fine-tuning del modelo para permitir múltiples marcas de agua y garantizar la imperceptibilidad.
- Licencia
- El modelo se ofrece bajo la licencia CC-BY-NC.
- Referencia
- Está relacionado con los proyectos AudioSeal y Segment Anything.
1 comentarios
Opiniones de Hacker News