[ADD][FEAT](loss) Add SigmoidCrossEntropyLoss

wbenbihi · Aug 23, 2022 · 7384c59 · 7384c59
1 parent 7c975af
commit 7384c59
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/hourglass_tensorflow/losses/__init__.py b/hourglass_tensorflow/losses/__init__.py
diff --git a/hourglass_tensorflow/losses/sigmoid_cross_entropy.py b/hourglass_tensorflow/losses/sigmoid_cross_entropy.py
@@ -0,0 +1,14 @@
+import tensorflow as tf
+import keras.losses
+
+
+class SigmoidCrossEntropyLoss(keras.losses.Loss):
+    def __init__(self, reduction=..., name=None, *args, **kwargs):
+        super().__init__(reduction, name)
+
+    def call(self, y_true, y_pred):
+        return tf.nn.sigmoid_cross_entropy_with_logits(
+            logits=y_pred,
+            labels=y_true,
+            name="nn.sigmoid_cross_entropy_with_logits",
+        )