Add reference

Author: qubvel

segmentation_models_pytorch/utils/functions.py

@@ -3,6 +3,8 @@
 
 def iou(pr, gt, eps=1e-7, threshold=None, activation='sigmoid'):
     """
+    Source:
+        https://github.com/catalyst-team/catalyst/
     Args:
         pr (torch.Tensor): A list of predicted elements
         gt (torch.Tensor):  A list of elements that are to be predicted
@@ -70,4 +72,4 @@ def f_score(pr, gt, beta=1, eps=1e-7, threshold=None, activation='sigmoid'):
     score = ((1 + beta ** 2) * tp + eps) \
             / ((1 + beta ** 2) * tp + beta ** 2 * fn + fp + eps)
 
-    return score
+    return score