ibm-nasa-geospatial
/

Prithvi-EO-1.0-100M-burn-scar

Image Segmentation

Foundation model

Model card Files Files and versions

Paolo-Fraccaro commited on Jul 31, 2023

Commit

fed16ce

·

1 Parent(s): 67b68ed

fix bug

Files changed (1) hide show

burn_scars_Prithvi_100M.py +1 -1

burn_scars_Prithvi_100M.py CHANGED Viewed

@@ -218,6 +218,6 @@ model = dict(
         align_corners=False,
         loss_decode=loss_func),
     train_cfg=dict(),
-    test_cfg=dict(mode='slide', stride=(tile_size/2, tile_size/2), crop_size=(tile_size, tile_size)))
 gpu_ids = range(0, 1)
 auto_resume = False

         align_corners=False,
         loss_decode=loss_func),
     train_cfg=dict(),
+    test_cfg=dict(mode='slide', stride=(int(tile_size/2), int(tile_size/2)), crop_size=(tile_size, tile_size)))
 gpu_ids = range(0, 1)
 auto_resume = False