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 |
|