Spaces:
Build error
Build error
Sophie98
commited on
Commit
•
14aef70
1
Parent(s):
2ca996f
fix error
Browse files- 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')
|