summaryrefslogtreecommitdiff
path: root/pkgs/development/python-modules/tensorflow-datasets/corruptions.patch
blob: 71e5da5e11bbda9c389505874783fc4493801252 (about) (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
diff --git a/tensorflow_datasets/image_classification/corruptions.py b/tensorflow_datasets/image_classification/corruptions.py
index 066c4460..cb9a6667 100644
--- a/tensorflow_datasets/image_classification/corruptions.py
+++ b/tensorflow_datasets/image_classification/corruptions.py
@@ -35,7 +35,7 @@ FROST_FILENAMES = []
 
 
 def _imagemagick_bin():
-  return 'imagemagick'  # pylint: disable=unreachable
+  return 'convert'  # pylint: disable=unreachable
 
 
 # /////////////// Corruption Helpers ///////////////
@@ -675,7 +675,7 @@ def spatter(x, severity=1):
     #     ker = np.array([[-1,-2,-3],[-2,0,0],[-3,0,1]], dtype=np.float32)
     #     ker -= np.mean(ker)
     ker = np.array([[-2, -1, 0], [-1, 1, 1], [0, 1, 2]])
-    dist = cv2.filter2D(dist, cv2.CVX_8U, ker)
+    dist = cv2.filter2D(dist, cv2.CV_8U, ker)
     dist = cv2.blur(dist, (3, 3)).astype(np.float32)
 
     m = cv2.cvtColor(liquid_layer * dist, cv2.COLOR_GRAY2BGRA)