mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
bb93b482b8
Thanks for your contribution and we appreciate it a lot. The following instructions would make your pull request more healthy and more easily get feedback. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers. ## Motivation Add Huasdorff distance loss --------- Co-authored-by: xiexinch <xiexinch@outlook.com>
21 lines
901 B
Python
21 lines
901 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .accuracy import Accuracy, accuracy
|
|
from .boundary_loss import BoundaryLoss
|
|
from .cross_entropy_loss import (CrossEntropyLoss, binary_cross_entropy,
|
|
cross_entropy, mask_cross_entropy)
|
|
from .dice_loss import DiceLoss
|
|
from .focal_loss import FocalLoss
|
|
from .huasdorff_distance_loss import HuasdorffDisstanceLoss
|
|
from .lovasz_loss import LovaszLoss
|
|
from .ohem_cross_entropy_loss import OhemCrossEntropy
|
|
from .tversky_loss import TverskyLoss
|
|
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', 'DiceLoss',
|
|
'FocalLoss', 'TverskyLoss', 'OhemCrossEntropy', 'BoundaryLoss',
|
|
'HuasdorffDisstanceLoss'
|
|
]
|