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
def filter_windows(windows): if scene.aoi_polygons: windows = Box.filter_by_aoi(windows, scene.aoi_polygons) return windows