def get_train_windows(self, scene):
     result = []
     extent = scene.raster_source.get_extent()
     chip_size = self.config.chip_size
     stride = chip_size
     windows = extent.get_windows(chip_size, stride)
     if scene.aoi_polygons:
         windows = Box.filter_by_aoi(windows, scene.aoi_polygons)
     for window in windows:
         chip = scene.raster_source.get_chip(window)
         if np.sum(chip.ravel()) > 0:
             result.append(window)
     return result
Ejemplo n.º 2
0
 def filter_windows(windows):
     if scene.aoi_polygons:
         windows = Box.filter_by_aoi(windows, scene.aoi_polygons)
     return windows