mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
0ee3016558
* add lovasz loss * Modify as comments * Modify paper url * add unittest and remove Var * impove unittest
12 lines
485 B
Python
12 lines
485 B
Python
from .accuracy import Accuracy, accuracy
|
|
from .cross_entropy_loss import (CrossEntropyLoss, binary_cross_entropy,
|
|
cross_entropy, mask_cross_entropy)
|
|
from .lovasz_loss import LovaszLoss
|
|
from .utils import reduce_loss, weight_reduce_loss, weighted_loss
|
|
|
|
__all__ = [
|
|
'accuracy', 'Accuracy', 'cross_entropy', 'binary_cross_entropy',
|
|
'mask_cross_entropy', 'CrossEntropyLoss', 'reduce_loss',
|
|
'weight_reduce_loss', 'weighted_loss', 'LovaszLoss'
|
|
]
|