xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix overlap_mask in SegmentationLoss (#2985)

Browse Source
This commit is contained in:
Laughing
2023-06-05 04:35:33 +08:00
committed by GitHub
parent bb5bbc2893
commit f502b50365
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ultralytics/yolo/utils/loss.py
View File

@ -178,10 +178,10 @@ class v8DetectionLoss:
# Criterion class for computing training losses # Criterion class for computing training losses
class v8SegmentationLoss(v8DetectionLoss): class v8SegmentationLoss(v8DetectionLoss):
def __init__(self, model, overlap=True): # model must be de-paralleled def __init__(self, model): # model must be de-paralleled
super().__init__(model) super().__init__(model)
self.nm = model.model[-1].nm # number of masks self.nm = model.model[-1].nm # number of masks
self.overlap = overlap self.overlap = model.args.overlap_mask
def __call__(self, preds, batch): def __call__(self, preds, batch):
"""Calculate and return the loss for the YOLO model.""" """Calculate and return the loss for the YOLO model."""