diff --git a/utils/raw_to_cog.py b/utils/raw_to_cog.py index 2d3075e..d65f17a 100644 --- a/utils/raw_to_cog.py +++ b/utils/raw_to_cog.py @@ -34,7 +34,7 @@ gdal.SetConfigOption("GDAL_CACHEMAX", "1024") def tif_to_cog( input_path: str, output_path: str, - output_type: gdal.GDT_DataType = gdal.GDT_Float32, + output_type: int = gdal.GDT_Float32, no_data: float = np.nan, compress: str = "DEFLATE", scaleParams: list = None, @@ -48,8 +48,8 @@ def tif_to_cog( 输入 GeoTIFF 文件路径 output_path : str 输出 COG 文件路径 - output_type : gdal.GDT_DataType, optional - 输出数据类型, 默认 float32, by default gdal.GDT_Float32 + output_type : int, optional + 输出数据类型, 本质是整数索引, 默认 float32, by default gdal.GDT_Float32 no_data : float, optional 无效值, 默认 np.nan, by default np.nan compress : str, optional @@ -116,6 +116,6 @@ if __name__ == "__main__": # 输入目录: 包含分块tif影像的根目录 input_root = Path(r"D:\CVEOdata\RS_Data\Terrain") file_name = "GLO30.DSM.2014.Hubei.30m.tif" - # 输出目录: 存放最终RGB镶嵌结果的目录 + # 输出目录: 存放最终COG结果的目录 output_root = Path(r"D:\CVEOdata\RS_Data\Terrain\COG") main(input_root, file_name, output_root)