python3Packages.mask-rcnn: fix for recent keras
This commit is contained in:
parent
4d8996155f
commit
d8edf418c1
|
@ -1,6 +1,7 @@
|
|||
{ buildPythonPackage
|
||||
, cython
|
||||
, fetchFromGitHub
|
||||
, fetchpatch
|
||||
, h5py
|
||||
, imgaug
|
||||
, ipython
|
||||
|
@ -26,6 +27,22 @@ buildPythonPackage rec {
|
|||
sha256 = "13s3q9yh2q9m9vyksd269mww3bni4q2w7q5l419q70ca075qp8zp";
|
||||
};
|
||||
|
||||
patches = [
|
||||
# Fix for TF2:
|
||||
# https://github.com/matterport/Mask_RCNN/issues/2734
|
||||
(fetchpatch {
|
||||
url = "https://github.com/BupyeongHealer/Mask_RCNN_tf_2.x/commit/7957839fe2b248f2f22c7e991ead12068ddc6cfc.diff";
|
||||
excludes = [ "mrcnn/model.py" ];
|
||||
sha256 = "sha256-70BGrx6X1uJDA2025f0YTlreT2uB3n35yIzuhf+ypVc=";
|
||||
})
|
||||
];
|
||||
|
||||
# Fix for recent Keras
|
||||
postPatch = ''
|
||||
substituteInPlace mrcnn/model.py \
|
||||
--replace "KE." "KL."
|
||||
'';
|
||||
|
||||
nativeBuildInputs = [ cython ];
|
||||
|
||||
propagatedBuildInputs = [
|
||||
|
|
Loading…
Reference in a new issue