readded seg objective #1

manole-alexandru · Mar 20, 2023 · ba46faf · ba46faf
1 parent de596c8
commit ba46faf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)