训练数据生成

在没有模型的情况下,可通过训练数据生成工具,基于原始遥感影像数据和矢量标注数据,生成模型训练需要使用的训练数据。由于遥感影像数据体量较大,且卷积神经网络对于运行内存的要求较高,因此需要先把影像处理成适合机器学习模型训练的数据。训练数据生成工具的目的就是把影像数据和标签数据同步切分成一张张小图片和标签,生成适合神经网络模型训练的数据。

一般情况下,不同的训练数据(如分辨率,观测高度、观测角度、对象间距等)对得到的模型精度有非常大的影响。训练数据选择一般有以下要求:数据样本尽可能大、数据多样化,数据样本质量较高。

数据准备

在生成训练数据之前,需要准备的影像数据,以及希望识别的地物所在具体位置的标签数据。例如要做建筑物的识别,就需要目标区域的影像数据和建筑物矢量轮廓的标签数据,基于以上的样本数据生成相应格式的训练数据。

  • 样本影像数据:目标区域的影像数据,建议使用分辨率较高的航空影像片和卫星影像片。
  • 样本标签数据:矢量面数据集,面对象为影像数据中对象目标的外廓多边形对象,及相关的属性信息,例如目标识别(飞机),标注信息需要标识是否是飞机、该飞机的像素坐标、飞机型号等信息。

训练数据生成

主要操作步骤如下:

  1. 功能入口工具箱->机器学习->影像分析->训练数据生成工具。
  2. 训练数据用途:不同用途生成的训练数据格式不同,提供了目标检测、二元分类、地物分类、场景分类、对象提取、通用变化检测六种用途。
  3. 文件类型:支持选择数据集和文件夹,可用于批量生成训练数据,默认值为数据集。训练数据用途选择通用变化检测时该参数不生效。
  4. 样本影像数据源/数据集文件类型选择数据集时填写该参数,选择需要生成训练数据的影像或镶嵌数据集,变化检测只支持输入影像数据集。
  5. 文件路径文件类型选择文件夹时填写该参数,选择需要生成训练数据的文件夹路径,会自动读取文件夹中*.tif、*.img 等格式的影像。。
  6. 样本标签数据源/数据集:在数据源中选择标签数据集所在的数据源,并在数据集处选择标签面数据集。
  7. 类别字段:设置标签数据中标识地物类型的字段。若矢量标签只有一类时,可设为空值。当训练数据用途选择二元分类或通用变化检测时该参数不生效。
  8. 行数/列数:生成训练数据的切片大小,行数和列数所填数值需一致,单元为像元(Pixel),默认值为1024px,图片大小建议为 64 的整数倍,其大小与训练占用内存成正比。
  9. 其他参数设置:勾选该复选框,可对以下图片格式、偏移量、索引起始值参数进行设置。
    • 图片格式:设置影像数据按照指定大小划分后的图像格式,提供与源数据相同、tiff、png、jpeg 四种图像格式。训练数据用途为目标检测时,图片仅支持 jpeg 格式。
    • 垂直/水平偏移:该值为重复切分时不同方向上偏移的大小,默认值为0。数据量较小时为了扩充训练数据并保证每个标签都可以完整的参与训练而不被切分,图片需要重复切片。如数据量足够大,该值可为0。
    • 索引起始值:系统会对切分生成的图片进行自然数顺序编码,默认从 00000000 开始,如用户需要进行特殊替换,则可自行设定起始值。默认值为 -1,表示如果选择的输出路径及数据名称文件夹下存在已经产生的训练数据,则默认向其中追加新生成的训练数据。若该值不为 -1,则在该文件夹下从设置的起始值开始生成数据,已有的同序号数据将被覆盖。
    • 保存无标签数据:勾选时可保存无标签的切片,用于补充负样本或平衡正负样本等情况。
  10. 对比影像:参与变化检测的对比影像,只在‘训练数据用途’为通用变化检测时此参数生效。支持输入影像数据集。
  11. 结果数据:设置结果数据的输出路径,及结果数据的文件夹名称。结果数据包括影像数据和标签数据切分的图像,及数据的配置文件(*.sda)。注意:输出路径不能带中文字符。

相关内容

机器学习环境配置

模型训练

目标检测

二元分类

地物分类

场景分类

对象提取

通用变化检测

对象提取