tjh.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4
5# dependencies
6, numpy
7, pillow
8, requests
9, scipy
10, torch
11, torchvision
12, tqdm
13}:
14
15buildPythonPackage rec {
16 pname = "clean-fid";
17 version = "0.1.35";
18 format = "setuptools";
19
20 src = fetchFromGitHub {
21 owner = "GaParmar";
22 repo = "clean-fid";
23 rev = "c8ffa420a3923e8fd87c1e75170de2cf59d2644b";
24 hash = "sha256-fqBU/TmCXDTPU3KTP0+VYQoP+HsT2UMcZeLzQHKD9hw=";
25 };
26
27 propagatedBuildInputs = [
28 numpy
29 pillow
30 requests
31 scipy
32 torch
33 torchvision
34 tqdm
35 ];
36
37 pythonImportsCheck = [
38 "cleanfid"
39 ];
40
41 # no tests1
42 doCheck = false;
43
44 meta = with lib; {
45 description = "PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]";
46 homepage = "https://github.com/GaParmar/clean-fid";
47 license = licenses.mit;
48 maintainers = teams.tts.members;
49 };
50}