Return False, None if the mask for a certain tile contains no valid p…

…ixels.
centreborelli · Mar 15, 2022 · 83acc2e · 83acc2e
1 parent 42dec23
commit 83acc2e
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: