pipelines

class aiearth.predict.pipelines.GeoDatasetProcessPipeline(processors: List[Callable], image_size: int = 1024, bound: int = 0)

基类：Pipeline

__call__(uri: str, out_dir: str = '.', out_filename: str = None, temp_dir: str = None, extent: Box | shapely.geometry.polygon.Polygon = None, aoi_shape_file: str = None, sink_raster_count: int = 1, mask_zeros: bool = False, chunk_size: int = 20000, polygonization=True, merged=True, dissolve_merged_boundary=True, max_num_pending_write: int = 100, raster_dataset_open_args: dict = {}, skip_errors: bool = False)

Call self as a function.

__init__(processors: List[Callable], image_size: int = 1024, bound: int = 0)

class aiearth.predict.pipelines.GeoSegmentationPredictPipeline(model_checkpoint, predictor_cls, predictor_num_gpus=1, predictor_memory=1073741824, predictor_batch_size=1, pre_processors: List[Callable] | None = None, post_processors: List[Callable] | None = None)

基类：Pipeline

__call__(uri: str, out_dir: str = '.', out_filename: str = None, temp_dir: str = None, extent: Box | shapely.geometry.polygon.Polygon = None, aoi_shape_file: str = None, sink_raster_count: int = 1, mask_zeros: bool = False, chunk_size: int = 20000, polygonization=True, merged=True, dissolve_merged_boundary=True, max_num_pending_write: int = 100, raster_dataset_open_args: dict = {}, skip_errors: bool = False)

Call self as a function.

__init__(model_checkpoint, predictor_cls, predictor_num_gpus=1, predictor_memory=1073741824, predictor_batch_size=1, pre_processors: List[Callable] | None = None, post_processors: List[Callable] | None = None)

class aiearth.predict.pipelines.Pipeline

基类：object

abstract __call__()

Call self as a function.

abstract __init__()