Spaces:

Chancee12
/

Satellite-Segmentation-AI

Build error

Chancee12 commited on Apr 14, 2023

Commit

5b0d644

1 Parent(s): 107eb45

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ dice_loss = sm.losses.DiceLoss(class_weights = weights)
 focal_loss = sm.losses.CategoricalFocalLoss()
 total_loss = dice_loss + (1 * focal_loss)
-satellite_model = load_model('model/satellite_segmentation_full_v2.h5', custom_objects=({'dice_loss_plus_1focal_loss' : total_loss, 'jaccard_coef': jaccard_coef}))
 def process_input_image(image_source):
     image = np.expand_dims(image_source, 0)

 focal_loss = sm.losses.CategoricalFocalLoss()
 total_loss = dice_loss + (1 * focal_loss)
+satellite_model = load_model('model/satellite_segmentation_v3_full', custom_objects=({'dice_loss_plus_1focal_loss' : total_loss, 'jaccard_coef': jaccard_coef}))
 def process_input_image(image_source):
     image = np.expand_dims(image_source, 0)