Skip to content

Commit e55b2da

Browse files
authored
Correct constant description (#335)
1 parent dcd19d6 commit e55b2da

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

segmentation_models_pytorch/losses/constants.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,15 @@
44
#: Target mask shape - (N, H, W), model output mask shape (N, 1, H, W).
55
BINARY_MODE: str = "binary"
66

7+
#: Loss multiclass mode suppose you are solving multi-**class** segmentation task.
8+
#: That mean you have *C = 1..N* classes which have unique label values,
9+
#: classes are mutually exclusive and all pixels are labeled with theese values.
10+
#: Target mask shape - (N, H, W), model output mask shape (N, C, H, W).
11+
MULTICLASS_MODE: str = "multiclass"
12+
713
#: Loss multilabel mode suppose you are solving multi-**label** segmentation task.
814
#: That mean you have *C = 1..N* classes which pixels are labeled as **1**,
915
#: classes are not mutually exclusive and each class have its own *channel*,
1016
#: pixels in each channel which are not belong to class labeled as **0**.
1117
#: Target mask shape - (N, C, H, W), model output mask shape (N, C, H, W).
12-
MULTICLASS_MODE: str = "multiclass"
13-
14-
#: Loss multiclass mode suppose you are solving multi-**class** segmentation task.
15-
#: That mean you have *C = 1..N* classes which have unique label values,
16-
#: classes are mutually exclusive and all pixels are labeled with theese values.
17-
#: Target mask shape - (N, H, W), model output mask shape (N, C, H, W).
1818
MULTILABEL_MODE: str = "multilabel"

0 commit comments

Comments
 (0)