Я пытаюсь использовать tf.boolean_mask для получения среднего замаскированного различия для сегментации изображения:
def custom_loss(image):
def loss(predicted_y, target_y):
pred_mask = tf.math.greater(predicted_y,0.5)
target_mask = tf.math.greater(target_y,0.5)
mean_diff = (tf.reduce_mean(tf.boolean_mask(image,pred_mask)) - tf.reduce_mean(tf.boolean_mask(image,target_mask))) ** 2
return mean_diff
return loss
К сожалению, я получаю ValueError: нет градиентов для какой-либо переменной, что логически может быть вызвано по tf.boolean_mask. Есть ли способ сделать это на Tensorflow 2.0?
Спасибо большое!