Skip to content

Instantly share code, notes, and snippets.

@stiv-yakovenko

stiv-yakovenko/bce_and_dice.py

Created June 12, 2019 22:25
Show Gist options
  • Save stiv-yakovenko/cb8ad7e9a2e922f23c44b04777c5aa02 to your computer and use it in GitHub Desktop.
Save stiv-yakovenko/cb8ad7e9a2e922f23c44b04777c5aa02 to your computer and use it in GitHub Desktop.
Download ZIP
bce+dice
Raw
bce_and_dice.py
smooth = 1.
intersection = tf.reduce_sum(flat_logits * flat_labels)
dice_score = (2 * intersection + smooth) / (
tf.reduce_sum(flat_labels) + tf.reduce_sum(flat_logits) + smooth)
dice_loss = 1 - dice_score
cross_entropy_loss = tf.reduce_mean(
tf.nn.softmax_cross_entropy_with_logits_v2(logits=flat_logits, labels=flat_labels))
loss = dice_loss + cross_entropy_loss
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment