深度学习预测结果拼接代码

最新推荐文章于 2024-04-12 19:54:58 发布

囹圄girl

最新推荐文章于 2024-04-12 19:54:58 发布

阅读量175

点赞数 4

文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/qq_45629716/article/details/137275807

版权

下面的代码是用来拼接预测之后的结果图代码，这是当时裁剪时和预测时均保留了坐标信息，所以在拼接结果时直接用osgeo库来拼接

from osgeo import gdal
import os
# 设置GDAL的环境变量，以便能找到影像文件的驱动
gdal.SetConfigOption('GDAL_FILENAME_IS_UTF8', 'YES')
# 影像文件所在的主文件夹路径
root_folder_path = r"/文件夹路径 /"
# 输出文件夹路径
output_folder_path = r"/文件夹路径 /"
# 遍历主文件夹下的所有子文件夹
for subdir in os.listdir(root_folder_path):
    subdir_path = os.path.join(root_folder_path, subdir)
    if os.path.isdir(subdir_path):
        # 获取子文件夹内所有影像文件的路径
        image_files = [os.path.join(subdir_path, f) for f in os.listdir(subdir_path) if f.endswith('.tif')]
        # 如果子文件夹内有影像文件，则进行拼接
        if image_files:
            # 拼接后的影像文件路径，以子文件夹名称命名
            output_file = os.path.join(output_folder_path, f"{subdir}.tif")
            # 使用gdal的warp工具来拼接影像
            gdal.Warp(output_file, image_files)