遥感数据处理#

SDK还提供了用于遥感数据处理的pipeline,这里以遥感RGB影像转灰度图为例进行流程说明

import依赖#

引入需要的包依赖

[ ]:
from aiearth.predict.processors import (
    Chw2HwcProcessor,
    RGB2GrayProcessor,
)
from aiearth.predict.pipelines import GeoDatasetProcessPipeline

from aiearth.predict.logging import root_logger as logger

初始化pipeline#

初始化遥感数据处理任务pipeline:

  • processors:处理算子,参数类型为List,可传递多个处理算子,在pipeline里面会将算子组合成一个调用链

[ ]:
pipe = GeoDatasetProcessPipeline(
    processors=[
        Chw2HwcProcessor(["image"]),
        RGB2GrayProcessor(["image"]),
    ]
)

运行pipeline#

运行pipeline,polygonization=False表示结果不进行转shape操作,pipeline运行完成之后会在当前目录输出结果tiff文件

[ ]:
logger.info("run pipeline")
pipe(
    "/path/to/your/tiff",
    polygonization=False,
)

运行的时候需要将/path/to/your/tiff路径替换为本地tiff影像的绝对路径