Spaces:
Build error
Build error
Sophie98 commited on
Commit ·
758d201
1
Parent(s): 5d930e1
typo
Browse files- segmentation.py +1 -1
segmentation.py
CHANGED
|
@@ -26,7 +26,7 @@ def get_mask(image:Image) -> Image:
|
|
| 26 |
|
| 27 |
# define network parameters
|
| 28 |
n_classes = 1 if len(CLASSES) == 1 else (len(CLASSES) + 1) # case for binary and multiclass segmentation
|
| 29 |
-
|
| 30 |
preprocess_input = sm.get_preprocessing(BACKBONE)
|
| 31 |
LR=0.0001
|
| 32 |
|
|
|
|
| 26 |
|
| 27 |
# define network parameters
|
| 28 |
n_classes = 1 if len(CLASSES) == 1 else (len(CLASSES) + 1) # case for binary and multiclass segmentation
|
| 29 |
+
activation = 'sigmoid' if n_classes == 1 else 'softmax'
|
| 30 |
preprocess_input = sm.get_preprocessing(BACKBONE)
|
| 31 |
LR=0.0001
|
| 32 |
|