srivatsavdamaraju
commited on
Update main.py
Browse files
main.py
CHANGED
@@ -25,7 +25,7 @@ def jacard_coef(y_true, y_pred):
|
|
25 |
intersection = K.sum(y_true_f * y_pred_f)
|
26 |
return (intersection + 1.0) / (K.sum(y_true_f) + K.sum(y_pred_f) - intersection + 1.0)
|
27 |
|
28 |
-
model_path =
|
29 |
custom_objects = {
|
30 |
"dice_loss_plus_1focal_loss": total_loss,
|
31 |
"jacard_coef": jacard_coef
|
|
|
25 |
intersection = K.sum(y_true_f * y_pred_f)
|
26 |
return (intersection + 1.0) / (K.sum(y_true_f) + K.sum(y_pred_f) - intersection + 1.0)
|
27 |
|
28 |
+
model_path = "satellite_standard_unet_100epochs.hdf5"
|
29 |
custom_objects = {
|
30 |
"dice_loss_plus_1focal_loss": total_loss,
|
31 |
"jacard_coef": jacard_coef
|