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)
|