Sophie98 commited on
Commit
758d201
1 Parent(s): 5d930e1
Files changed (1) hide show
  1. 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
- actvation = 'sigmoid' if n_classes == 1 else 'softmax'
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