mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-06-30 21:15:21 +03:00
23 lines
943 B
Diff
23 lines
943 B
Diff
|
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)
|