From f145e9d33d52c4b98ac66caff5a167cf8d0f9e6f Mon Sep 17 00:00:00 2001 From: venkatram-dev Date: Tue, 20 Aug 2024 12:46:58 -0700 Subject: [PATCH 1/2] fix_box_loss_error --- roboflow/core/version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/roboflow/core/version.py b/roboflow/core/version.py index 0723d2be..6c5221f2 100644 --- a/roboflow/core/version.py +++ b/roboflow/core/version.py @@ -413,7 +413,7 @@ def live_plot(epochs, mAP, loss, title=""): mAP = np.array([float(epoch["mAP"]) for epoch in models["roboflow-train"]["epochs"]]) loss = np.array( [ - (float(epoch["box_loss"]) + float(epoch["class_loss"]) + float(epoch["obj_loss"])) + sum(float(epoch[key]) for key in ['box_loss', 'class_loss', 'obj_loss'] if key in epoch) for epoch in models["roboflow-train"]["epochs"] ] ) From c4d65acffcfa505a5173e0d2ac73b90e3aea0b06 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 20 Aug 2024 19:59:16 +0000 Subject: [PATCH 2/2] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20?= =?UTF-8?q?format=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- roboflow/core/version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/roboflow/core/version.py b/roboflow/core/version.py index 6c5221f2..1f8f4de7 100644 --- a/roboflow/core/version.py +++ b/roboflow/core/version.py @@ -413,7 +413,7 @@ def live_plot(epochs, mAP, loss, title=""): mAP = np.array([float(epoch["mAP"]) for epoch in models["roboflow-train"]["epochs"]]) loss = np.array( [ - sum(float(epoch[key]) for key in ['box_loss', 'class_loss', 'obj_loss'] if key in epoch) + sum(float(epoch[key]) for key in ["box_loss", "class_loss", "obj_loss"] if key in epoch) for epoch in models["roboflow-train"]["epochs"] ] )