diff --git a/pkgs/development/python-modules/invisible-watermark/default.nix b/pkgs/development/python-modules/invisible-watermark/default.nix new file mode 100644 index 000000000000..82357741bab5 --- /dev/null +++ b/pkgs/development/python-modules/invisible-watermark/default.nix @@ -0,0 +1,49 @@ +{ lib +, buildPythonPackage +, pythonOlder +, fetchFromGitHub +, opencv4 +, pytorch +, onnx +, onnxruntime +, pillow +, pywavelets +, numpy +}: + +buildPythonPackage rec { + pname = "invisible-watermark"; + version = "0.1.5"; + disabled = pythonOlder "3.6"; + + src = fetchFromGitHub { + owner = "ShieldMnt"; + repo = "invisible-watermark"; + rev = version; + sha256 = "sha256-NGDPEETuM7rYbo8kXYoRWLJWpa/lWLKEvaaiDzSWYZ4="; + }; + + propagatedBuildInputs = [ + opencv4 + pytorch + onnx + onnxruntime + pillow + pywavelets + numpy + ]; + + postPatch = '' + substituteInPlace setup.py \ + --replace 'opencv-python>=4.1.0.25' 'opencv' + ''; + + pythonImportsCheck = [ "imwatermark" ]; + + meta = with lib; { + description = "A library for creating and decoding invisible image watermarks"; + homepage = "https://github.com/ShieldMnt/invisible-watermark"; + license = licenses.mit; + maintainers = with maintainers; [ Luflosi ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 5e26e836eb87..0b9ac35ef5fb 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -4565,6 +4565,8 @@ self: super: with self; { intreehooks = callPackage ../development/python-modules/intreehooks { }; + invisible-watermark = callPackage ../development/python-modules/invisible-watermark { }; + invocations = callPackage ../development/python-modules/invocations { }; invoke = callPackage ../development/python-modules/invoke { };