🐛 [Fix] Some display bug, enable print, log

yolo/config/general.yaml

@@ -9,7 +9,7 @@ out_path: runs
 exist_ok: True
 
 lucky_number: 10
-use_wandb: True
+use_wandb: False
 use_TensorBoard: False
 
 weight: True # Path to weight or True for auto, False for no pretrained weight

yolo/tools/solver.py

@@ -122,6 +122,7 @@ class ModelTrainer:
             self.progress.finish_one_epoch(epoch_loss, epoch)
 
             self.validator.solve(self.validation_dataloader, epoch_idx=epoch)
+        self.progress.finish_train()
 
 
 class ModelTester:

yolo/utils/bounding_box_utils.py

@@ -268,7 +268,7 @@ class Vec2Box:
     def __init__(self, model: YOLO, image_size, device):
         self.device = device
 
-        if getattr(model, "strides"):
+        if hasattr(model, "strides"):
             logger.info(f"🈶 Found stride of model {model.strides}")
             self.strides = model.strides
         else:

yolo/utils/solver_utils.py

@@ -1,5 +1,5 @@
-import os
-import sys
+import contextlib
+import io
 
 from pycocotools.coco import COCO
 from pycocotools.cocoeval import COCOeval
@@ -7,13 +7,12 @@ from rich.table import Table
 
 
 def calculate_ap(coco_gt: COCO, pd_path):
-    sys.stdout = open(os.devnull, "w")
-    coco_dt = coco_gt.loadRes(pd_path)
-    coco_eval = COCOeval(coco_gt, coco_dt, "bbox")
-    coco_eval.evaluate()
-    coco_eval.accumulate()
-    coco_eval.summarize()
-    sys.stdout = sys.__stdout__
+    with contextlib.redirect_stdout(io.StringIO()):
+        coco_dt = coco_gt.loadRes(pd_path)
+        coco_eval = COCOeval(coco_gt, coco_dt, "bbox")
+        coco_eval.evaluate()
+        coco_eval.accumulate()
+        coco_eval.summarize()
     return coco_eval.stats