__init__.py 210 Bytes
Newer Older
mibaumgartner's avatar
losses  
mibaumgartner committed
1
2
3
from nndet.losses.classification import focal_loss_with_logits, FocalLossWithLogits
from nndet.losses.regression import SmoothL1Loss, smooth_l1_loss, GIoULoss
from nndet.losses.segmentation import SoftDiceLoss