diff --git a/maintainers/maintainer-list.nix b/maintainers/maintainer-list.nix index e168e7913842..adbfa42991e8 100644 --- a/maintainers/maintainer-list.nix +++ b/maintainers/maintainer-list.nix @@ -12076,6 +12076,11 @@ githubId = 1198065; name = "Jeffrey David Johnson"; }; + jeffcshelton = { + github = "jeffcshelton"; + githubId = 77426854; + name = "Jeff Shelton"; + }; jefferyoo = { email = "oojefferywm@proton.me"; github = "jefferyoo"; diff --git a/pkgs/development/python-modules/lpips/default.nix b/pkgs/development/python-modules/lpips/default.nix new file mode 100644 index 000000000000..5e52af91d503 --- /dev/null +++ b/pkgs/development/python-modules/lpips/default.nix @@ -0,0 +1,49 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + numpy, + scipy, + setuptools, + torch, + torchvision, + tqdm, +}: + +buildPythonPackage (finalAttrs: { + pname = "lpips"; + version = "0.1.4"; + pyproject = true; + + src = fetchFromGitHub { + owner = "richzhang"; + repo = "PerceptualSimilarity"; + tag = "v${finalAttrs.version}"; + hash = "sha256-dIQ9B/HV/2kUnXLXNxAZKHmv/Xv37kl2n6+8IfwIALE="; + }; + + build-system = [ + setuptools + ]; + + dependencies = [ + numpy + scipy + torch + torchvision + tqdm + ]; + + # Tests require network access to download pretrained models. + doCheck = false; + + pythonImportsCheck = [ "lpips" ]; + + meta = { + description = "Learned Perceptual Image Patch Similarity metric"; + homepage = "https://github.com/richzhang/PerceptualSimilarity"; + changelog = "https://github.com/richzhang/PerceptualSimilarity/releases/tag/${finalAttrs.src.tag}"; + license = lib.licenses.bsd2; + maintainers = with lib.maintainers; [ jeffcshelton ]; + }; +}) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index ba7043d0f868..9817c7e27545 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -9180,6 +9180,8 @@ self: super: with self; { lpc-checksum = callPackage ../development/python-modules/lpc-checksum { }; + lpips = callPackage ../development/python-modules/lpips { }; + lrcalc-python = callPackage ../development/python-modules/lrcalc-python { }; lrclibapi = callPackage ../development/python-modules/lrclibapi { };