├── .gitignore ├── LICENSE ├── README.md ├── datasets ├── README.md ├── convert_water_vit_to_wds.py └── watermark_ds_v1_aligned_url_cap.tar.gz └── training ├── eval_watermark.py ├── scalable_shampoo ├── README.md ├── __init__.py ├── matrix_functions.py ├── shampoo.py └── shampoo_utils.py └── train_watermark.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/README.md -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/convert_water_vit_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/datasets/convert_water_vit_to_wds.py -------------------------------------------------------------------------------- /datasets/watermark_ds_v1_aligned_url_cap.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/datasets/watermark_ds_v1_aligned_url_cap.tar.gz -------------------------------------------------------------------------------- /training/eval_watermark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/training/eval_watermark.py -------------------------------------------------------------------------------- /training/scalable_shampoo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/training/scalable_shampoo/README.md -------------------------------------------------------------------------------- /training/scalable_shampoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/training/scalable_shampoo/__init__.py -------------------------------------------------------------------------------- /training/scalable_shampoo/matrix_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/training/scalable_shampoo/matrix_functions.py -------------------------------------------------------------------------------- /training/scalable_shampoo/shampoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/training/scalable_shampoo/shampoo.py -------------------------------------------------------------------------------- /training/scalable_shampoo/shampoo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/training/scalable_shampoo/shampoo_utils.py -------------------------------------------------------------------------------- /training/train_watermark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LAION-AI/watermark-detection/HEAD/training/train_watermark.py --------------------------------------------------------------------------------