YOLOP
YOLOP copied to clipboard
It is mentioned in the paper that Lcls and Lobj in the detection task use focal loss, but this is not the case in the code.
in the code loss.py, line 195:
gamma = cfg.LOSS.FL_GAMMA if gamma > 0: BCEcls, BCEobj = FocalLoss(BCEcls, gamma), FocalLoss(BCEobj, gamma)
but in the code default.py, line 39:
_C.LOSS.FL_GAMMA = 0.0
Following the code above, Lcls and Lobj are still BCELoss.
没开启