Removed prints

manole-alexandru · Apr 8, 2023 · cd1ffae · cd1ffae
1 parent a2a2b1f
commit cd1ffae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -532,7 +532,7 @@ def parse_opt(known=False):
 
 
 def main(opt, callbacks=Callbacks()):
-    print('\n---------- VERSION:', '#0010', '----------\n')
+    print('\n---------- VERSION:', '#0011', '----------\n')
     # Checks
     if RANK in {-1, 0}:
         print_args(vars(opt))

diff --git a/utils/plots.py b/utils/plots.py
@@ -240,7 +240,7 @@ def plot_masks(segs, seg_preds, fname='mask.jpg', fname_real='real.jpg'):
     bs, _, h, w = segs.shape  # batch size, _, height, width
     bs = min(bs, max_subplots)  # limit plot images
     ns = np.ceil(bs ** 0.5)  # number of subplots (square)
-    print('\n--------------NS: ', ns, '------------\n')
+    # print('\n--------------NS: ', ns, '------------\n')
 
     mosaic = np.full((int(ns * h), int(ns * w)), 255, dtype=np.uint8)  # init
     for i, seg_pred in enumerate(seg_preds):