forked from mirrors/nixpkgs
b2737d4980
* python3Packages.tensorflow-datasets: init at 4.4.0 * Update pkgs/development/python-modules/tensorflow-datasets/default.nix Co-authored-by: Dmitry Kalinkin <dmitry.kalinkin@gmail.com> Co-authored-by: Samuel Ainsworth <skainsworth@gmail.com> Co-authored-by: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
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)
|