diff --git a/utils/raw_to_cog.py b/utils/raw_to_cog.py index 2179f56..eaa133f 100644 --- a/utils/raw_to_cog.py +++ b/utils/raw_to_cog.py @@ -8,7 +8,7 @@ COG (Cloud Optimized GeoTIFF) 是一种优化设计的 GeoTIFF 文件格式, 特 ------------------------------------------------------------------------------- Authors: CVEO Team -Last Updated: 2026-01-08 +Last Updated: 2026-01-28 =============================================================================== """ @@ -76,6 +76,8 @@ def tif_to_cog( no_data = -32768 elif output_type == gdal.GDT_Float32: no_data = -9999 + elif output_type == gdal.GDT_Byte: + no_data = None else: no_data = np.nan logging.warning(f"原数据未设无效值, 现已设为: {no_data}") @@ -113,8 +115,12 @@ def main(input_dir, file_name, output_path, output_type=gdal.GDT_Float32): input_path = input_dir / file_name output_path = output_dir / file_name # 关于无效值 - # Float32 类型可以设为 -9999; 8bit 类型可以设为 0 - tif_to_cog(str(input_path), str(output_path), output_type, no_data=-9999) + # Float32 类型可以设为 -9999 + if output_type == gdal.GDT_Float32: + no_data = -9999 + else: + no_data = np.nan + tif_to_cog(str(input_path), str(output_path), output_type, no_data=no_data) if __name__ == "__main__":