Python遥感影像深度学习指南(1)-使用卷积神经网络(CNN、U-Net)和 FastAI进行简单云层检测

        【遥感影像深度学习】系列的第一章,Python遥感影像深度学习的入门课程,介绍如何使用卷积神经网络(CNN)从卫星图像中分割云层

1、数据集

        在本项目中,我们将使用 Kaggle 提供的 38-Cloud Segmentation in Satellite Images数据集。

        该数据集由裁剪成 384x384 (适用于深度学习)的卫星场景组成。总共有 8400 个用于训练和 9201 个用于测试的图块,分别位于红、绿、蓝和近红外(近红外)波段的目录中,还有一个额外的目录用于存储参考掩膜(真实值 - *_gt)。结构如图 2 所示。 

2、数据预处理

        为了给神经网络准备好数据,我们首先需要对数据进行预处理,以适合我们的模型。由于我们将使用预训练的 ResNet 框架,因此模型中只能拟合 3 个波段,最好是 RGB 波段,因为模型已经用这些波段进行了预训练。因此,数据预处理的第一步就是从给定的图像中创建 RGB patch小块。我们将使用 PIL(Python Imaging Library,Python 图像库)打开红、绿、蓝 .tif 图像,并将其保存为单一的 RGB .png 文件。下面的代码就可以做到这一点(确保与原始数据集的结构相同),并对图像进行归一化处理。

from PIL import Image
from pathlib import Path
import numpy as np

def create_rgb_pil (red_filename: Path):
    # set the filenames for each patch
    red_fn = str(red_filename)
    blue_fn = red_fn.replace('red','blue')
    green_fn = red_fn.replace('red', 'green')
    rgb_fn = red_fn.replace('red', 'rgb').replace('.TIF', '.png')
    
    # open red, green and blue patches as numpy arrays
    arr_red = np.array(Image.open(red_fn))
    arr_green = np.array(Image.open(green_fn))
    arr_blue = np.array(Image.open(blue_f

相关推荐

  1. 当代深度学习模型介绍--神经网络CNNs

    2023-12-24 16:34:03       40 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2023-12-24 16:34:03       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2023-12-24 16:34:03       101 阅读
  3. 在Django里面运行非项目文件

    2023-12-24 16:34:03       82 阅读
  4. Python语言-面向对象

    2023-12-24 16:34:03       91 阅读

热门阅读

  1. WordBoggle字谜问题算法实现

    2023-12-24 16:34:03       57 阅读
  2. 77. 组合

    2023-12-24 16:34:03       55 阅读
  3. css基础回顾2

    2023-12-24 16:34:03       53 阅读
  4. MySQL-2

    MySQL-2

    2023-12-24 16:34:03      54 阅读
  5. 『OPEN3D』1.7 点云拟合问题

    2023-12-24 16:34:03       77 阅读
  6. 在线分析工具-日志优化

    2023-12-24 16:34:03       60 阅读
  7. winlogbeat收集Windows事件日志传给ELK

    2023-12-24 16:34:03       52 阅读
  8. Spring的API获取泛型实际类型

    2023-12-24 16:34:03       73 阅读
  9. 基于蚁群算法的臂凿岩机器人动态孔序规划

    2023-12-24 16:34:03       56 阅读
  10. 聊聊equals()方法

    2023-12-24 16:34:03       58 阅读
  11. VSC(Visual Studio Code)好用插件推荐

    2023-12-24 16:34:03       62 阅读