You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
943 B
22 lines
943 B
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)
|
|
|