diff --git a/pkgs/default.nix b/pkgs/default.nix index 4771653..2dca311 100644 --- a/pkgs/default.nix +++ b/pkgs/default.nix @@ -6,6 +6,10 @@ inherit (pkgs) lib fetchPypi; python310Packages = pkgs.python310.pkgs; }; + python-opencv-headless = pkgs.callPackage ./python-opencv-headless { + inherit (pkgs) lib fetchPypi; + python310Packages = pkgs.python310.pkgs; + }; in { fish-rust = pkgs.callPackage ./fish-rust {}; python-pymatting = python-pymatting; diff --git a/pkgs/python-opencv-headless/default.nix b/pkgs/python-opencv-headless/default.nix new file mode 100644 index 0000000..2ac0c2c --- /dev/null +++ b/pkgs/python-opencv-headless/default.nix @@ -0,0 +1,34 @@ +{ lib +, python310Packages +, fetchPypi +}: + +python310Packages.buildPythonPackage rec { + pname = "opencv-python-headless"; + version = "4.9.0.80"; + format = "wheel"; + + src = fetchPypi { + inherit pname version format; + dist = "cp310"; + abi = "cp310"; + platform = "manylinux_2_17_x86_64.manylinux2014_x86_64"; + hash = "sha256-a1uEcgIUu7qxcJCyX+4umIOGa4bsRdfgSBn6jO0PnoM="; + }; + + propagatedBuildInputs = with python310Packages; [ + numpy + ]; + + doCheck = false; + + pythonImportsCheck = [ "cv2" ]; + + meta = with lib; { + description = "Wrapper package for OpenCV python bindings"; + homepage = "https://github.com/opencv/opencv-python"; + license = licenses.mit; + platforms = platforms.linux; + maintainers = with maintainers; [ "Cobray" ]; + }; +} diff --git a/pkgs/python-rembg/default.nix b/pkgs/python-rembg/default.nix index 97ac0d8..d1c9751 100644 --- a/pkgs/python-rembg/default.nix +++ b/pkgs/python-rembg/default.nix @@ -3,6 +3,7 @@ python310Packages, fetchPypi, python-pymatting, + python-opencv-headless, }: python310Packages.buildPythonPackage rec { pname = "rembg"; @@ -23,7 +24,7 @@ python310Packages.buildPythonPackage rec { numpy pillow onnxruntime - opencv4 + python-opencv-headless requests aiohttp asynctest @@ -47,6 +48,6 @@ python310Packages.buildPythonPackage rec { homepage = "https://github.com/danielgatis/rembg"; license = licenses.mit; platforms = platforms.all; - maintainers = with maintainers; [ "Cobray" ]; + maintainers = with maintainers; ["Cobray"]; }; }