├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── export_onnx.py ├── imwatermark ├── __init__.py ├── dwtDctSvd.py ├── maxDct.py ├── rivaGan.py ├── rivagan_decoder.onnx ├── rivagan_encoder.onnx └── watermark.py ├── invisible-watermark ├── pypi.sh ├── setup.py ├── test_vectors ├── original.jpg └── peppers.png └── tests ├── __init__.py ├── test_watermarking.py └── test_wavelets.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/README.md -------------------------------------------------------------------------------- /export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/export_onnx.py -------------------------------------------------------------------------------- /imwatermark/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/imwatermark/__init__.py -------------------------------------------------------------------------------- /imwatermark/dwtDctSvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/imwatermark/dwtDctSvd.py -------------------------------------------------------------------------------- /imwatermark/maxDct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/imwatermark/maxDct.py -------------------------------------------------------------------------------- /imwatermark/rivaGan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/imwatermark/rivaGan.py -------------------------------------------------------------------------------- /imwatermark/rivagan_decoder.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/imwatermark/rivagan_decoder.onnx -------------------------------------------------------------------------------- /imwatermark/rivagan_encoder.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/imwatermark/rivagan_encoder.onnx -------------------------------------------------------------------------------- /imwatermark/watermark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/imwatermark/watermark.py -------------------------------------------------------------------------------- /invisible-watermark: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/invisible-watermark -------------------------------------------------------------------------------- /pypi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/pypi.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/setup.py -------------------------------------------------------------------------------- /test_vectors/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/test_vectors/original.jpg -------------------------------------------------------------------------------- /test_vectors/peppers.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/test_vectors/peppers.png -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_watermarking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/tests/test_watermarking.py -------------------------------------------------------------------------------- /tests/test_wavelets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stability-AI/invisible-watermark-gpu/HEAD/tests/test_wavelets.py --------------------------------------------------------------------------------