nix-dotfiles/pkgs/python-rembg/default.nix
2025-03-17 16:10:17 -06:00

49 lines
895 B
Nix

{
lib,
python310Packages,
fetchPypi,
python-pymatting,
}:
python310Packages.buildPythonPackage rec {
pname = "rembg";
version = "2.0.50";
format = "setuptools";
src = fetchPypi {
inherit pname version;
sha256 = "0dgq291bj4w6jkcyz7lvp1vba2nczfnmxa2acl2sqib5p8cpzjvc";
};
nativeBuildInputs = with python310Packages; [
setuptools
poetry-core
];
propagatedBuildInputs = with python310Packages; [
numpy
pillow
onnxruntime
opencv4
requests
aiohttp
asynctest
click
filetype
pooch
pympler
python-pymatting
scikit-image
scipy
tqdm
];
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"];
};
}