From ba46faf8c65e831cf79c0efbcba63ecfd10e07a1 Mon Sep 17 00:00:00 2001 From: ManoleAlexandru99 Date: Mon, 20 Mar 2023 13:47:35 +0200 Subject: [PATCH] readded seg objective #0001 --- utils/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loss.py b/utils/loss.py index 132763cedd34..c4e6bd6ec448 100644 --- a/utils/loss.py +++ b/utils/loss.py @@ -186,7 +186,7 @@ def __call__(self, preds, targets, seg_masks): # predictions, targets # return (lbox + lobj + lcls) * bs, torch.cat((lbox, lobj, lcls)).detach() # return total_loss, torch.cat((lbox, lobj, lcls, lseg)).detach() - return (lbox + lobj + lcls) * bs * 1, lseg * bs * 0, torch.cat((lbox, lobj, lcls, lseg)).detach() + return (lbox + lobj + lcls) * bs * 1, lseg * bs * 0.2, torch.cat((lbox, lobj, lcls, lseg)).detach() def build_targets(self, p, targets): # Build targets for compute_loss(), input targets(image,class,x,y,w,h)