Seg Test ultralytics#7

manole-alexandru · Mar 25, 2023 · b5f321e · b5f321e
1 parent 8a37b49
commit b5f321e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/val.py b/val.py
@@ -314,6 +314,7 @@ def run(
     pf = '%22s' + '%11i' * 2 + '%11.3g' * 6  # print format
     iou = iou / len(pbar)
     rail_iou = rail_iou / len(pbar)
+    print('\n----- RAIL_IOU:', rail_iou, '-------\n')
     LOGGER.info(pf % ('all', seen, nt.sum(), mp, mr, map50, map, iou, rail_iou))
     if nt.sum() == 0:
         LOGGER.warning(f'WARNING ⚠️ no labels found in {task} set, can not compute metrics without labels')