Enabled multitask with weights 1, 1 ultralytics#5

manole-alexandru · Mar 23, 2023 · a93dfbf · a93dfbf
1 parent bef3e29
commit a93dfbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/loss.py b/utils/loss.py
@@ -185,7 +185,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 * 0, lseg * bs * 1, torch.cat((lbox, lobj, lcls, lseg)).detach()
+        return (lbox + lobj + lcls) * bs * 1, lseg * bs * 1, 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)