too lazy to write but it works

This commit is contained in:
alsaiduq-lab 2025-03-22 22:33:01 -06:00
parent 4b536b9b11
commit 3a7ec9f943
6 changed files with 145 additions and 90 deletions

View file

@ -1,44 +1,44 @@
{ lib,
python311Packages,
fetchPypi,
python-pymatting,
python-opencv-headless
{ lib
, python311Packages
, fetchPypi
, python-pymatting
, python-opencv-headless
}:
python311Packages.buildPythonPackage rec {
pname = "rembg";
version = "2.0.50";
format = "pyproject";
src = fetchPypi {
inherit pname version;
hash = "sha256-bMt/GbplRawFZUqoXq37zAq1dribnu/ZlIYTuUIS+DU=";
sha256 = "bMt/GbplRawFZUqoXq37zAq1dribnu/ZlIYTuUIS+DU=";
};
nativeBuildInputs = with python311Packages; [
poetry-core
setuptools
];
propagatedBuildInputs = with python311Packages; [
numpy
onnxruntime
python-opencv-headless
pillow
pooch
python-pymatting
scikit-image
scipy
tqdm
aiohttp
pytorch-bin
] ++ [
python-opencv-headless
python-pymatting
];
pythonRemoveDeps = [ "opencv-python-headless" ];
dontPrecompilePages = true;
doInstallCheck = false;
doCheck = false;
meta = with lib; {
description = "Tool to remove image backgrounds";
homepage = "https://github.com/danielgatis/rembg";
license = licenses.mit;
platforms = platforms.all;
maintainers = with maintainers; ["Cobray"];
maintainers = with maintainers; [ "Cobray" ];
};
}