mmpretrain/mmcls/models/losses/__init__.py

15 lines
671 B
Python
Raw Normal View History

from .accuracy import Accuracy, accuracy
from .cross_entropy_loss import (CrossEntropyLoss, binary_cross_entropy,
cross_entropy)
from .eval_metrics import f1_score, precision, recall
from .focal_loss import FocalLoss, sigmoid_focal_loss
from .label_smooth_loss import LabelSmoothLoss, label_smooth
2020-07-08 21:45:44 +08:00
from .utils import reduce_loss, weight_reduce_loss, weighted_loss
__all__ = [
'accuracy', 'Accuracy', 'cross_entropy', 'binary_cross_entropy',
'CrossEntropyLoss', 'reduce_loss', 'weight_reduce_loss', 'label_smooth',
'LabelSmoothLoss', 'weighted_loss', 'precision', 'recall', 'f1_score',
'FocalLoss', 'sigmoid_focal_loss'
]