pyrox.dev
1{ buildPythonPackage
2, cython
3, fetchFromGitHub
4, h5py
5, imgaug
6, ipython
7, Keras
8, lib
9, matplotlib
10, numpy
11, opencv3
12, pillow
13, scikitimage
14, scipy
15, tensorflow_2
16}:
17
18buildPythonPackage rec {
19 pname = "mask-rcnn";
20 version = "2.1";
21
22 src = fetchFromGitHub {
23 owner = "matterport";
24 repo = "Mask_RCNN";
25 rev = "3deaec5d902d16e1daf56b62d5971d428dc920bc";
26 sha256 = "13s3q9yh2q9m9vyksd269mww3bni4q2w7q5l419q70ca075qp8zp";
27 };
28
29 nativeBuildInputs = [ cython ];
30
31 propagatedBuildInputs = [
32 h5py
33 imgaug
34 ipython
35 Keras
36 matplotlib
37 numpy
38 opencv3
39 pillow
40 scikitimage
41 scipy
42 tensorflow_2 # Keras only supports tensorflow 2 now
43 ];
44
45 meta = with lib; {
46 description = "Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow";
47 homepage = "https://github.com/matterport/Mask_RCNN";
48 license = licenses.mit;
49 maintainers = with maintainers; [ rakesh4g ];
50 };
51}