Sophie98 commited on
Commit
14aef70
1 Parent(s): 2ca996f
Files changed (1) hide show
  1. segmentation.py +1 -0
segmentation.py CHANGED
@@ -31,6 +31,7 @@ def get_mask(image):
31
  # total_loss = sm.losses.binary_focal_dice_loss # or sm.losses.categorical_focal_dice_loss
32
  metrics = [sm.metrics.IOUScore(threshold=0.5), sm.metrics.FScore(threshold=0.5)]
33
  # compile keras model with defined optimozer, loss and metrics
 
34
  model.compile(optim, total_loss, metrics)
35
 
36
  print('loading model weights')
 
31
  # total_loss = sm.losses.binary_focal_dice_loss # or sm.losses.categorical_focal_dice_loss
32
  metrics = [sm.metrics.IOUScore(threshold=0.5), sm.metrics.FScore(threshold=0.5)]
33
  # compile keras model with defined optimozer, loss and metrics
34
+ print('compiling')
35
  model.compile(optim, total_loss, metrics)
36
 
37
  print('loading model weights')