pyrox.dev
1{ lib
2, buildPythonPackage
3, isPy27
4, fetchFromGitHub
5, pytestCheckHook
6, unittest2
7, future
8, numpy
9, pillow
10, scipy
11, scikit-learn
12, scikitimage
13, threadpoolctl
14}:
15
16buildPythonPackage rec {
17 pname = "batchgenerators";
18 version = "0.21";
19
20 disabled = isPy27;
21
22 src = fetchFromGitHub {
23 owner = "MIC-DKFZ";
24 repo = pname;
25 rev = "v${version}";
26 sha256 = "16bk4r0q3m2c9fawpmj4l7kz0x3fyv1spb92grf44gmyricq3jdb";
27
28 };
29
30 propagatedBuildInputs = [
31 future numpy pillow scipy scikit-learn scikitimage threadpoolctl
32 ];
33
34 checkInputs = [ pytestCheckHook unittest2 ];
35
36 meta = {
37 description = "2D and 3D image data augmentation for deep learning";
38 homepage = "https://github.com/MIC-DKFZ/batchgenerators";
39 license = lib.licenses.asl20;
40 maintainers = with lib.maintainers; [ bcdarwin ];
41 };
42}