diff --git a/utils/loss.py b/utils/loss.py index c4e6bd6ec448..ae53ad096d25 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.2, torch.cat((lbox, lobj, lcls, lseg)).detach() + return (lbox + lobj + lcls) * bs * 0, 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)