nur-packages

commit abfd221c12daf678ebf829fe96acd038de188aea
parent ce246ce5d2badafae06920bad311651691fb6ffd
Author: Nikolay Korotkiy <sikmir@disroot.org>
Date:   Sat, 15 Aug 2026 17:13:21 +0400

Add segmentation-models-pytorch

Diffstat:
A
pkgs/by-name/se/segmentation-models-pytorch/package.nix
 | 
48
++++++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 48 insertions(+), 0 deletions(-)
diff --git a/pkgs/by-name/se/segmentation-models-pytorch/package.nix b/pkgs/by-name/se/segmentation-models-pytorch/package.nix
@@ -0,0 +1,48 @@
+{
+  lib,
+  fetchFromGitHub,
+  python3Packages,
+  writableTmpDirAsHomeHook,
+}:
+
+python3Packages.buildPythonPackage (finalAttrs: {
+  pname = "segmentation-models-pytorch";
+  version = "0.5.0";
+  pyproject = true;
+
+  __structuredAttrs = true;
+
+  src = fetchFromGitHub {
+    owner = "qubvel-org";
+    repo = "segmentation_models.pytorch";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-QtrmMbVcFHftV69stJHk0+3n1o0inlO22xHs/smLlGg=";
+  };
+
+  build-system = with python3Packages; [ setuptools ];
+
+  dependencies = with python3Packages; [
+    numpy
+    pillow
+    safetensors
+    timm
+    torch
+    torchvision
+    tqdm
+  ];
+
+  nativeCheckInputs = with python3Packages; [
+    pytestCheckHook
+    gitpython
+    writableTmpDirAsHomeHook
+  ];
+
+  doCheck = false;
+
+  meta = {
+    description = "Image segmentation models with pre-trained backbones";
+    homepage = "https://github.com/qubvel-org/segmentation_models.pytorch";
+    license = lib.licenses.mit;
+    maintainers = [ lib.maintainers.sikmir ];
+  };
+})