Merge pull request #8 from lapix-ufsc/improve_metrics_models

Improve metrics models
lapix-ufsc · Jan 3, 2022 · 5944f5b · 5944f5b
2 parents f471dfe + c4646d1
commit 5944f5b
Show file tree

Hide file tree

Showing 5 changed files with 190 additions and 56 deletions.
diff --git a/lapixdl/evaluation/evaluate.py b/lapixdl/evaluation/evaluate.py
@@ -52,7 +52,7 @@ def evaluate_segmentation(gt_masks: Iterable[Mask],
 
     metrics = SegmentationMetrics(classes, confusion_matrix)
 
-    print(metrics)
+    print(metrics.to_dataframe())
 
     return metrics
 
@@ -83,7 +83,7 @@ def evaluate_classification(gt_classifications: Iterable[Classification],
 
     metrics = ClassificationMetrics(classes, confusion_matrix)
 
-    print(metrics)
+    print(metrics.to_dataframe())
 
     return metrics
 
@@ -174,7 +174,7 @@ def evaluate_detection(gt_bboxes: Iterable[List[BBox]],
         cls_ious_sum / tot_images,
         predictions_by_class)
 
-    print(metrics)
+    print(metrics.to_dataframe())
 
     return metrics