pyrox.dev
1{ buildPythonPackage
2, fetchFromGitHub
3, pythonOlder
4, pytestCheckHook
5, albumentations
6, dill
7, h5py
8, hickle
9, numpy
10, opencv4
11, pandas
12, pillow
13, prodict
14, pycm
15, pyyaml
16, scipy
17, requests
18, scikitimage
19, six
20, tabulate
21, torch
22, torchvision
23, tqdm
24, yacs
25, lib
26}:
27
28buildPythonPackage rec {
29 pname = "pywick";
30 version = "0.6.5";
31
32 disabled = pythonOlder "3.6";
33
34 src = fetchFromGitHub {
35 owner = "achaiah";
36 repo = pname;
37 rev = "v${version}";
38 sha256 = "0wnijdvqgdpzfdsy1cga3bsr0n7zzsl8hp4dskqwxx087g5h1r84";
39 };
40
41 postPatch = ''
42 substituteInPlace setup.py --replace "opencv-python-headless" "opencv"
43 '';
44
45 propagatedBuildInputs = [
46 albumentations
47 dill
48 h5py
49 hickle
50 numpy
51 opencv4
52 pandas
53 pillow
54 prodict
55 pycm
56 pyyaml
57 scipy
58 requests
59 scikitimage
60 tabulate
61 torch
62 torchvision
63 tqdm
64 six
65 yacs
66 ];
67
68 checkInputs = [ pytestCheckHook ];
69
70 meta = {
71 description = "High-level training framework for Pytorch";
72 homepage = "https://github.com/achaiah/pywick";
73 license = lib.licenses.mit;
74 maintainers = with lib.maintainers; [ bcdarwin ];
75 };
76}