diff options
author | happysalada <raphael@megzari.com> | 2023-07-09 10:55:08 +0800 |
---|---|---|
committer | Yt <happysalada@tuta.io> | 2023-07-09 13:09:04 +0800 |
commit | 1f2493e5b668c0ffb505ce77ee15bb2c79da943b (patch) | |
tree | e5f9a9b61e933677cb4356a91899c6d206781d5f /pkgs/development/python-modules/effdet | |
parent | e978b715242fe780fa065e0414a434578a101a71 (diff) |
python310Packages.effdet: init at 0.4.1
Diffstat (limited to 'pkgs/development/python-modules/effdet')
-rw-r--r-- | pkgs/development/python-modules/effdet/default.nix | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/effdet/default.nix b/pkgs/development/python-modules/effdet/default.nix new file mode 100644 index 0000000000000..e6a7945ae0be6 --- /dev/null +++ b/pkgs/development/python-modules/effdet/default.nix @@ -0,0 +1,43 @@ +{ lib +, buildPythonPackage +, fetchPypi +# build inputs +, torch +, torchvision +, timm +, pycocotools +, omegaconf +}: +let + pname = "effdet"; + version = "0.4.1"; +in +buildPythonPackage { + inherit pname version; + format = "setuptools"; + + src = fetchPypi { + inherit pname version; + hash = "sha256-rFWJ/TBKVlDCAZhrLvX44QwREJOnGxxJ+muIF3EIErU="; + }; + + propagatedBuildInputs = [ + torch + torchvision + timm + pycocotools + omegaconf + ]; + + # Project has no tests + doCheck = false; + + pythonImportsCheck = [ "effdet" ]; + + meta = { + description = "A PyTorch implementation of EfficientDet"; + homepage = "https://pypi.org/project/effdet"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ happysalada ]; + }; +} |