Merge pull request #313 from venkatram-dev/fix_box_loss_error

grzegorz-roboflow · web-flow · commit db89776a7138 · 2024-09-17T09:16:53.000+02:00
fix_box_loss_error
diff --git 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"]
                     ]
                 )

Original file line number	Diff line number	Diff line change
`@@ -413,7 +413,7 @@ def live_plot(epochs, mAP, loss, title=""):`
`413`	`413`	`mAP = np.array([float(epoch["mAP"]) for epoch in models["roboflow-train"]["epochs"]])`
`414`	`414`	`loss = np.array(`
`415`	`415`	`[`
`416`		`- (float(epoch["box_loss"]) + float(epoch["class_loss"]) + float(epoch["obj_loss"]))`
	`416`	`+ sum(float(epoch[key]) for key in ["box_loss", "class_loss", "obj_loss"] if key in epoch)`
`417`	`417`	`for epoch in models["roboflow-train"]["epochs"]`
`418`	`418`	`]`
`419`	`419`	`)`