make cutout op compatible with non eager mode (#2190)

rybakov · web-flow · commit cc621dcfee0f · 2020-10-06T11:25:25.000+02:00
cutout op is not compatible with non eager mode, this is a fix
diff --git a/tensorflow_addons/image/cutout_ops.py b/tensorflow_addons/image/cutout_ops.py
@@ -189,7 +189,7 @@ def cutout(
             mask_4d = tf.expand_dims(masks.stack(), 1)
             mask = tf.tile(mask_4d, [1, tf.shape(images)[1], 1, 1])
         images = tf.where(
-            mask == 0,
+            tf.equal(mask, 0),
             tf.ones_like(images, dtype=images.dtype) * constant_values,
             images,
         )

Original file line number	Diff line number	Diff line change
`@@ -189,7 +189,7 @@ def cutout(`
`189`	`189`	`mask_4d = tf.expand_dims(masks.stack(), 1)`
`190`	`190`	`mask = tf.tile(mask_4d, [1, tf.shape(images)[1], 1, 1])`
`191`	`191`	`images = tf.where(`
`192`		`- mask == 0,`
	`192`	`+ tf.equal(mask, 0),`
`193`	`193`	`tf.ones_like(images, dtype=images.dtype) * constant_values,`
`194`	`194`	`images,`
`195`	`195`	`)`