Update boundary_loss.py

Author: huixiancheng

modules/losses/boundary_loss.py

@@ -55,14 +55,14 @@ def forward(self, pred, gt):
         pred_b -= 1 - pred
 
         # Visualization Boundary
-        for i in range(c):
-            gt_bv = gt_b.detach().cpu().numpy()
-            # cv2.imshow('gt_b_cls.png'.format(i), gt_bv[0][i])
-            cv2.imwrite('gt_b_cls{}.png'.format(i), gt_bv[0][i]*255)
-
-            pred_bv = pred_b.detach().cpu().numpy()
-            #cv2.imshow('pred_b_cls{}'.format(i), pred_bv[0][i])
-            cv2.imwrite('pred_b_cls{}.png'.format(i), pred_bv[0][i]*255)
+        # for i in range(c):
+        #     gt_bv = gt_b.detach().cpu().numpy()
+        #     # cv2.imshow('gt_b_cls.png'.format(i), gt_bv[0][i])
+        #     cv2.imwrite('gt_b_cls{}.png'.format(i), gt_bv[0][i]*255)
+        # 
+        #     pred_bv = pred_b.detach().cpu().numpy()
+        #     #cv2.imshow('pred_b_cls{}'.format(i), pred_bv[0][i])
+        #     cv2.imwrite('pred_b_cls{}.png'.format(i), pred_bv[0][i]*255)
 
 
         # reshape