From 0817e37e27226e1b1e1c26c9c9f64ec9c6d5a634 Mon Sep 17 00:00:00 2001 From: sakaguchi050403 Date: Tue, 13 Feb 2024 16:58:10 +0900 Subject: [PATCH 1/2] Add Accuracy metrics log to std output --- utils/loggers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 6a5673052521..a5561d35e5f9 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -227,7 +227,7 @@ def on_fit_epoch_end(self, vals, epoch, best_fitness, fi): x = dict(zip(self.keys, vals)) print( - f"Epoch: {epoch}, Training Loss: {vals[0].cpu().numpy().tolist()}, Validation Loss: {vals[7]}" + f"Epoch: {epoch}, Training Loss: {vals[0].cpu().numpy().tolist()}, Validation Loss: {vals[7]}, Validation Precision: {vals[3]}, Validation Recall: {vals[4]}, Validation mAP_0.5: {vals[5]}, Validation metrics/mAP_0.5:0.95: {vals[6]}" ) if self.csv: From de7af6639a07cc0c8f81bfe246b09b83de433180 Mon Sep 17 00:00:00 2001 From: sakaguchi050403 Date: Thu, 21 Mar 2024 20:59:09 +0900 Subject: [PATCH 2/2] Change accuracy metrics. --- utils/loggers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index a5561d35e5f9..283c9d59a549 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -227,7 +227,7 @@ def on_fit_epoch_end(self, vals, epoch, best_fitness, fi): x = dict(zip(self.keys, vals)) print( - f"Epoch: {epoch}, Training Loss: {vals[0].cpu().numpy().tolist()}, Validation Loss: {vals[7]}, Validation Precision: {vals[3]}, Validation Recall: {vals[4]}, Validation mAP_0.5: {vals[5]}, Validation metrics/mAP_0.5:0.95: {vals[6]}" + f"Epoch: {epoch}, Training Loss: {vals[0].cpu().numpy().tolist()}, Validation Loss: {vals[7]}, Validation mAP_0.5: {vals[5]}, Validation mAP_0.5:0.95: {vals[6]}" ) if self.csv: