dataset#
- aiearth.predict.dataset.read_raster_dataset(raster_dataset: RasterDataset, image_size, bound, block_size=1, block_memory_buffer=100, aoi_vector_dataset: VectorDataset | None = None, read_block_image_column='image', read_block_box_column='box', read_block_pad_bounds_column='pad_bounds', size_bytes_growth_factor=1.25, ray_remote_args: Dict[str, Any] | None = None, **read_args)[源代码]#
datasource#
- class aiearth.predict.dataset.datasource.RasterDatasetReader(raster_dataset: RasterDataset, image_size, bound, block_size, block_memory_buffer, aoi_vector_dataset: VectorDataset, read_block_image_column, read_block_box_column, read_block_pad_bounds_column, size_bytes_growth_factor)[源代码]#
基类：
Reader- estimate_inmemory_data_size() int | None[源代码]#
Return an estimate of the in-memory data size, or None if unknown.
Note that the in-memory data size may be larger than the on-disk data size.
- get_read_tasks(parallelism: int) List[ReadTask][源代码]#
Execute the read and return read tasks.
- 参数:
parallelism – The requested read parallelism. The number of read tasks should equal to this value if possible.
read_args – Additional kwargs to pass to the datasource impl.
- 返回:
A list of read tasks that can be executed to read blocks from the datasource in parallel.
- class aiearth.predict.dataset.datasource.RasterDatasource[源代码]#
基类：
Datasource- create_reader(raster_dataset, image_size, bound, block_size, block_memory_buffer, aoi_vector_dataset, read_block_image_column, read_block_box_column, read_block_pad_bounds_column, size_bytes_growth_factor, **read_args) Reader[T][源代码]#
Return a Reader for the given read arguments.
The reader object will be responsible for querying the read metadata, and generating the actual read tasks to retrieve the data blocks upon request.
- 参数:
read_args – Additional kwargs to pass to the datasource impl.
raster#
- class aiearth.predict.dataset.raster.AIEGeoserverRasterDataset(uri, extent: Box | None = None, open_args={}, read_args={})[源代码]#