Deepwatermap深度学习项目跑自己的遥感影像提取水体

1、尊重Github作者，先放链接

 

isikdogan/deepwatermap: a deep model that segments water on multispectral images

2、如何跑自己的遥感影像

2.1、提取项目所需的6个波段

上篇文章介绍了这个开源代码程序，跑的是示例影像，下面介绍如何跑自己的影像。

只要满足自己输入的遥感影像波段满足上述即可，输入图像的分辨率无要求，只要GPU够大。

从所需波段可以看出这大致是一个landsat8的影响集，所以这边我们去地理空间数据云也下载一个

地理空间数据云

我这里所下载的是青岛市黄岛区的影像

下面做波段提取，代码如下

import os

import rasterio

from rasterio.merge import merge

import numpy as np



def get_band_files_from_folder(folder_path):

    # 搜索文件夹中的文件，匹配包含 B2, B3, B4, B5, B6, B7 的 TIF 文件

    band_files = {}

    for root, dirs, files in os.walk(folder_path):

        for file in files:

            if file.endswith('.tif') or file.endswith('.TIF'):

                if 'B2' in file:

                    band_files[2] = os.path.join(root, file)

                elif &#