Update Active-Contour-Loss.py

lc82111 · web-flow · commit 5cf9785d0dbb · 2019-10-20T15:41:43.000+08:00
diff --git a/Active-Contour-Loss.py b/Active-Contour-Loss.py
@@ -3,7 +3,7 @@
 
 def active_contour_loss(y_true, y_pred, weight=10):
   '''
-  y_true, y_pred: tensor of shape (B, C, H, W)
+  y_true, y_pred: tensor of shape (B, C, H, W), where y_true[:,:,region_in_contour] == 1, y_true[:,:,region_out_contour] == 0.
   weight: scalar, length term weight.
   '''
   # length term
