Merge pull request #116 from daviddemeij/masking

Avoid computing masked tiles
centreborelli · Mar 17, 2022 · ca589ef · ca589ef
2 parents a1daee5 + 83acc2e
commit ca589ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/s2p/initialization.py b/s2p/initialization.py
@@ -308,6 +308,8 @@ def is_this_tile_useful(x, y, w, h, images_sizes):
     wat_msk = cfg['images'][0]['wat']
     mask = masking.image_tile_mask(x, y, w, h, roi_msk, cld_msk, wat_msk,
                                    images_sizes[0], cfg['border_margin'])
+    if not mask.any():
+        return False, None
     return True, mask
 
 

diff --git a/s2p/masking.py b/s2p/masking.py
@@ -69,6 +69,8 @@ def image_tile_mask(x, y, w, h, roi_gml=None, cld_gml=None, raster_mask=None,
         with rasterio.open(raster_mask, 'r') as f:
             mask = np.logical_and(mask, f.read(window=((y, y+h), (x, x+w)),
                                                boundless=True).squeeze())
+        if not mask.any():
+            return mask
 
     # image borders mask
     if img_shape is not None: