Skip to content

Commit cf71646

Browse files
committed
Initialized Focal Tversky loss
1 parent 8147853 commit cf71646

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

segmentation_models_pytorch/losses/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from .soft_ce import SoftCrossEntropyLoss
99
from .tversky import TverskyLoss
1010
from .mcc import MCCLoss
11+
from .focal_tversky import FocalTverskyLoss
1112

1213
__all__ = [
1314
"BINARY_MODE",
@@ -21,4 +22,5 @@
2122
"SoftCrossEntropyLoss",
2223
"TverskyLoss",
2324
"MCCLoss",
25+
"FocalTverskyLoss",
2426
]

0 commit comments

Comments
 (0)