Merge pull request #268619 from tweag/lib-descriptions
[NixPkgs.git] / pkgs / development / python-modules / tensorflow-datasets / corruptions.patch
blob71e5da5e11bbda9c389505874783fc4493801252
1 diff --git a/tensorflow_datasets/image_classification/corruptions.py b/tensorflow_datasets/image_classification/corruptions.py
2 index 066c4460..cb9a6667 100644
3 --- a/tensorflow_datasets/image_classification/corruptions.py
4 +++ b/tensorflow_datasets/image_classification/corruptions.py
5 @@ -35,7 +35,7 @@ FROST_FILENAMES = []
8 def _imagemagick_bin():
9 - return 'imagemagick' # pylint: disable=unreachable
10 + return 'convert' # pylint: disable=unreachable
13 # /////////////// Corruption Helpers ///////////////
14 @@ -675,7 +675,7 @@ def spatter(x, severity=1):
15 # ker = np.array([[-1,-2,-3],[-2,0,0],[-3,0,1]], dtype=np.float32)
16 # ker -= np.mean(ker)
17 ker = np.array([[-2, -1, 0], [-1, 1, 1], [0, 1, 2]])
18 - dist = cv2.filter2D(dist, cv2.CVX_8U, ker)
19 + dist = cv2.filter2D(dist, cv2.CV_8U, ker)
20 dist = cv2.blur(dist, (3, 3)).astype(np.float32)
22 m = cv2.cvtColor(liquid_layer * dist, cv2.COLOR_GRAY2BGRA)