├── .github ├── CODE_OF_CONDUCT.md └── CONTRIBUTING.md ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── assets ├── .DS_Store ├── imgs │ └── 1.jpg └── videos │ └── 1.mp4 ├── configs ├── ablations │ ├── embedder.yaml │ └── extractor.yaml ├── all_augs.yaml ├── attenuation.yaml ├── datasets │ ├── coco.yaml │ ├── sa-1b-full-resized.yaml │ ├── sa-1b.yaml │ └── sa-v.yaml ├── embedder.yaml └── extractor.yaml ├── docs ├── baselines.md ├── torchscript.md ├── training.md └── vmaf.md ├── inference_av.py ├── inference_streaming.py ├── notebooks ├── colab.ipynb ├── image_inference.ipynb ├── video_inference.ipynb └── video_inference_streaming.ipynb ├── pyproject.toml ├── requirements.txt ├── scripts ├── inflate_model_to_temporal.py ├── make_config.py └── process_sab.py ├── train.py ├── videoseal ├── __init__.py ├── augmentation │ ├── __init__.py │ ├── augmenter.py │ ├── doaugs.py │ ├── geometric.py │ ├── masks.py │ ├── sequential.py │ ├── valuemetric.py │ └── video.py ├── cards │ ├── videoseal_0.0.yaml │ └── videoseal_1.0.yaml ├── data │ ├── __init__.py │ ├── datasets.py │ ├── loader.py │ └── transforms.py ├── evals │ ├── __init__.py │ ├── flops.py │ ├── full.py │ ├── metrics.py │ ├── speed.py │ └── step_size_eval.py ├── losses │ ├── __init__.py │ ├── dists.py │ ├── focal.py │ ├── jndloss.py │ ├── perceptual.py │ ├── ssim.py │ ├── videosealloss.py │ ├── watson_fft.py │ ├── watson_vgg.py │ └── yuvloss.py ├── models │ ├── __init__.py │ ├── baselines.py │ ├── blender.py │ ├── embedder.py │ ├── extractor.py │ ├── videoseal.py │ └── wam.py ├── modules │ ├── __init__.py │ ├── common.py │ ├── convnext.py │ ├── discriminator.py │ ├── dvmark.py │ ├── hidden.py │ ├── jnd.py │ ├── msg_processor.py │ ├── pixel_decoder.py │ ├── unet.py │ ├── vae.py │ └── vit.py └── utils │ ├── __init__.py │ ├── cfg.py │ ├── data.py │ ├── display.py │ ├── dist.py │ ├── image.py │ ├── logger.py │ ├── optim.py │ └── tensorboard.py └── wmforger ├── README.md ├── assets ├── tahiti.png └── tahiti_watermarked.png ├── configs ├── all_augs_v3.yaml ├── datasets │ └── sa-1b-full.yaml └── extractor.yaml ├── example.ipynb ├── optimize_image.py ├── train-slurm.sh ├── train.py └── wmforger ├── __init__.py ├── augmentation ├── __init__.py ├── augmenter.py ├── geometric.py ├── masks.py ├── sequential.py ├── valuemetric.py └── video.py ├── data ├── __init__.py ├── loader.py └── transforms.py ├── models ├── __init__.py └── extractor.py ├── modules ├── __init__.py ├── common.py ├── convnext.py ├── pixel_decoder.py ├── vit.py └── watermark_generators.py └── utils ├── __init__.py ├── data.py ├── dist.py ├── logger.py ├── optim.py └── tensorboard.py /.github/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/.github/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/README.md -------------------------------------------------------------------------------- /assets/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/assets/.DS_Store -------------------------------------------------------------------------------- /assets/imgs/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/assets/imgs/1.jpg -------------------------------------------------------------------------------- /assets/videos/1.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/assets/videos/1.mp4 -------------------------------------------------------------------------------- /configs/ablations/embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/ablations/embedder.yaml -------------------------------------------------------------------------------- /configs/ablations/extractor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/ablations/extractor.yaml -------------------------------------------------------------------------------- /configs/all_augs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/all_augs.yaml -------------------------------------------------------------------------------- /configs/attenuation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/attenuation.yaml -------------------------------------------------------------------------------- /configs/datasets/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/datasets/coco.yaml -------------------------------------------------------------------------------- /configs/datasets/sa-1b-full-resized.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/datasets/sa-1b-full-resized.yaml -------------------------------------------------------------------------------- /configs/datasets/sa-1b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/datasets/sa-1b.yaml -------------------------------------------------------------------------------- /configs/datasets/sa-v.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/datasets/sa-v.yaml -------------------------------------------------------------------------------- /configs/embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/embedder.yaml -------------------------------------------------------------------------------- /configs/extractor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/configs/extractor.yaml -------------------------------------------------------------------------------- /docs/baselines.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/docs/baselines.md -------------------------------------------------------------------------------- /docs/torchscript.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/docs/torchscript.md -------------------------------------------------------------------------------- /docs/training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/docs/training.md -------------------------------------------------------------------------------- /docs/vmaf.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/docs/vmaf.md -------------------------------------------------------------------------------- /inference_av.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/inference_av.py -------------------------------------------------------------------------------- /inference_streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/inference_streaming.py -------------------------------------------------------------------------------- /notebooks/colab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/notebooks/colab.ipynb -------------------------------------------------------------------------------- /notebooks/image_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/notebooks/image_inference.ipynb -------------------------------------------------------------------------------- /notebooks/video_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/notebooks/video_inference.ipynb -------------------------------------------------------------------------------- /notebooks/video_inference_streaming.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/notebooks/video_inference_streaming.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/inflate_model_to_temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/scripts/inflate_model_to_temporal.py -------------------------------------------------------------------------------- /scripts/make_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/scripts/make_config.py -------------------------------------------------------------------------------- /scripts/process_sab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/scripts/process_sab.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/train.py -------------------------------------------------------------------------------- /videoseal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/__init__.py -------------------------------------------------------------------------------- /videoseal/augmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/__init__.py -------------------------------------------------------------------------------- /videoseal/augmentation/augmenter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/augmenter.py -------------------------------------------------------------------------------- /videoseal/augmentation/doaugs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/doaugs.py -------------------------------------------------------------------------------- /videoseal/augmentation/geometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/geometric.py -------------------------------------------------------------------------------- /videoseal/augmentation/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/masks.py -------------------------------------------------------------------------------- /videoseal/augmentation/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/sequential.py -------------------------------------------------------------------------------- /videoseal/augmentation/valuemetric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/valuemetric.py -------------------------------------------------------------------------------- /videoseal/augmentation/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/augmentation/video.py -------------------------------------------------------------------------------- /videoseal/cards/videoseal_0.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/cards/videoseal_0.0.yaml -------------------------------------------------------------------------------- /videoseal/cards/videoseal_1.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/cards/videoseal_1.0.yaml -------------------------------------------------------------------------------- /videoseal/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /videoseal/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/data/datasets.py -------------------------------------------------------------------------------- /videoseal/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/data/loader.py -------------------------------------------------------------------------------- /videoseal/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/data/transforms.py -------------------------------------------------------------------------------- /videoseal/evals/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /videoseal/evals/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/evals/flops.py -------------------------------------------------------------------------------- /videoseal/evals/full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/evals/full.py -------------------------------------------------------------------------------- /videoseal/evals/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/evals/metrics.py -------------------------------------------------------------------------------- /videoseal/evals/speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/evals/speed.py -------------------------------------------------------------------------------- /videoseal/evals/step_size_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/evals/step_size_eval.py -------------------------------------------------------------------------------- /videoseal/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/__init__.py -------------------------------------------------------------------------------- /videoseal/losses/dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/dists.py -------------------------------------------------------------------------------- /videoseal/losses/focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/focal.py -------------------------------------------------------------------------------- /videoseal/losses/jndloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/jndloss.py -------------------------------------------------------------------------------- /videoseal/losses/perceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/perceptual.py -------------------------------------------------------------------------------- /videoseal/losses/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/ssim.py -------------------------------------------------------------------------------- /videoseal/losses/videosealloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/videosealloss.py -------------------------------------------------------------------------------- /videoseal/losses/watson_fft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/watson_fft.py -------------------------------------------------------------------------------- /videoseal/losses/watson_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/watson_vgg.py -------------------------------------------------------------------------------- /videoseal/losses/yuvloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/losses/yuvloss.py -------------------------------------------------------------------------------- /videoseal/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/models/__init__.py -------------------------------------------------------------------------------- /videoseal/models/baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/models/baselines.py -------------------------------------------------------------------------------- /videoseal/models/blender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/models/blender.py -------------------------------------------------------------------------------- /videoseal/models/embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/models/embedder.py -------------------------------------------------------------------------------- /videoseal/models/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/models/extractor.py -------------------------------------------------------------------------------- /videoseal/models/videoseal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/models/videoseal.py -------------------------------------------------------------------------------- /videoseal/models/wam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/models/wam.py -------------------------------------------------------------------------------- /videoseal/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/__init__.py -------------------------------------------------------------------------------- /videoseal/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/common.py -------------------------------------------------------------------------------- /videoseal/modules/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/convnext.py -------------------------------------------------------------------------------- /videoseal/modules/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/discriminator.py -------------------------------------------------------------------------------- /videoseal/modules/dvmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/dvmark.py -------------------------------------------------------------------------------- /videoseal/modules/hidden.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/hidden.py -------------------------------------------------------------------------------- /videoseal/modules/jnd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/jnd.py -------------------------------------------------------------------------------- /videoseal/modules/msg_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/msg_processor.py -------------------------------------------------------------------------------- /videoseal/modules/pixel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/pixel_decoder.py -------------------------------------------------------------------------------- /videoseal/modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/unet.py -------------------------------------------------------------------------------- /videoseal/modules/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/vae.py -------------------------------------------------------------------------------- /videoseal/modules/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/modules/vit.py -------------------------------------------------------------------------------- /videoseal/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/__init__.py -------------------------------------------------------------------------------- /videoseal/utils/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/cfg.py -------------------------------------------------------------------------------- /videoseal/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/data.py -------------------------------------------------------------------------------- /videoseal/utils/display.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/display.py -------------------------------------------------------------------------------- /videoseal/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/dist.py -------------------------------------------------------------------------------- /videoseal/utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/image.py -------------------------------------------------------------------------------- /videoseal/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/logger.py -------------------------------------------------------------------------------- /videoseal/utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/optim.py -------------------------------------------------------------------------------- /videoseal/utils/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/videoseal/utils/tensorboard.py -------------------------------------------------------------------------------- /wmforger/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/README.md -------------------------------------------------------------------------------- /wmforger/assets/tahiti.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/assets/tahiti.png -------------------------------------------------------------------------------- /wmforger/assets/tahiti_watermarked.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/assets/tahiti_watermarked.png -------------------------------------------------------------------------------- /wmforger/configs/all_augs_v3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/configs/all_augs_v3.yaml -------------------------------------------------------------------------------- /wmforger/configs/datasets/sa-1b-full.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/configs/datasets/sa-1b-full.yaml -------------------------------------------------------------------------------- /wmforger/configs/extractor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/configs/extractor.yaml -------------------------------------------------------------------------------- /wmforger/example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/example.ipynb -------------------------------------------------------------------------------- /wmforger/optimize_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/optimize_image.py -------------------------------------------------------------------------------- /wmforger/train-slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/train-slurm.sh -------------------------------------------------------------------------------- /wmforger/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/train.py -------------------------------------------------------------------------------- /wmforger/wmforger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/__init__.py -------------------------------------------------------------------------------- /wmforger/wmforger/augmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/augmentation/__init__.py -------------------------------------------------------------------------------- /wmforger/wmforger/augmentation/augmenter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/augmentation/augmenter.py -------------------------------------------------------------------------------- /wmforger/wmforger/augmentation/geometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/augmentation/geometric.py -------------------------------------------------------------------------------- /wmforger/wmforger/augmentation/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/augmentation/masks.py -------------------------------------------------------------------------------- /wmforger/wmforger/augmentation/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/augmentation/sequential.py -------------------------------------------------------------------------------- /wmforger/wmforger/augmentation/valuemetric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/augmentation/valuemetric.py -------------------------------------------------------------------------------- /wmforger/wmforger/augmentation/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/augmentation/video.py -------------------------------------------------------------------------------- /wmforger/wmforger/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/data/__init__.py -------------------------------------------------------------------------------- /wmforger/wmforger/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/data/loader.py -------------------------------------------------------------------------------- /wmforger/wmforger/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/data/transforms.py -------------------------------------------------------------------------------- /wmforger/wmforger/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/models/__init__.py -------------------------------------------------------------------------------- /wmforger/wmforger/models/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/models/extractor.py -------------------------------------------------------------------------------- /wmforger/wmforger/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/modules/__init__.py -------------------------------------------------------------------------------- /wmforger/wmforger/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/modules/common.py -------------------------------------------------------------------------------- /wmforger/wmforger/modules/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/modules/convnext.py -------------------------------------------------------------------------------- /wmforger/wmforger/modules/pixel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/modules/pixel_decoder.py -------------------------------------------------------------------------------- /wmforger/wmforger/modules/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/modules/vit.py -------------------------------------------------------------------------------- /wmforger/wmforger/modules/watermark_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/modules/watermark_generators.py -------------------------------------------------------------------------------- /wmforger/wmforger/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/utils/__init__.py -------------------------------------------------------------------------------- /wmforger/wmforger/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/utils/data.py -------------------------------------------------------------------------------- /wmforger/wmforger/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/utils/dist.py -------------------------------------------------------------------------------- /wmforger/wmforger/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/utils/logger.py -------------------------------------------------------------------------------- /wmforger/wmforger/utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/utils/optim.py -------------------------------------------------------------------------------- /wmforger/wmforger/utils/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/videoseal/HEAD/wmforger/wmforger/utils/tensorboard.py --------------------------------------------------------------------------------