您可以根据自己的地理区域、影像数据或感兴趣要素,对“建筑物覆盖区提取 - 澳大利亚”模型进行个性化微调。 与训练全新模型相比,微调现有模型所需的训练数据量更少,且节省计算资源和时间。
如 ArcGIS 提供的预训练深度学习模型未能达到预期效果,我们推荐您尝试对模型进行微调。 当您的感兴趣区域超出模型的适用地理范围,或者您的影像数据在分辨率、比例尺、季节性等属性上有所差异时,这一做法尤其适用。
您可以借助导出训练数据进行深度学习工具来准备训练数据。 然后,您可以利用 ArcGIS Pro 中的训练深度学习模型工具在数据集上对模型进行微调。 按照以下步骤微调模型。
准备训练数据
该模型训练的基础为三波段 RGB 影像,并结合了建筑物覆盖区标注。 利用导出训练数据进行深度学习工具,为模型微调准备训练数据。
- 浏览至分析选项卡下的工具。
- 单击地理处理窗格中的工具箱选项卡,选择影像分析工具,然后浏览至深度学习工具集下的导出训练数据进行深度学习工具。
- 按照以下说明设置参数选项卡下的变量:
- 对于输入栅格,选择三波段 RGB 影像。
- 对于输出文件夹,浏览至您选择的计算机上的任一目录。
- 可以为输入要素类、分类栅格或表选择包含建筑物覆盖区训练标注的标注要素类。
- 对于类值字段,设置引用了上述要素类中类值(“1”)的 ClassValue 字段。
- 对于图像格式,选择 TIFF 格式。
- 对于切片大小 X,输入 512。
- 对于切片大小 Y,输入512。
- 对于步幅 X,输入 0。
- 对于步幅 Y,输入 0。
- 对于元数据格式,选择 RCNN 掩膜。
- 设置环境选项卡下的变量。
- 对于处理范围,根据需要从下拉菜单中选择当前显示范围或任意其他选项。
- 对于像元大小,将值设置为所需的像元大小。
- 单击运行。
处理完成后,导出的训练数据将存储在指定目录中。
微调“北极海豹检测”模型
按照以下步骤微调模型:
- 浏览至分析选项卡下的工具。
- 在地理处理窗格中,单击工具箱,然后展开影像分析工具。 选择深度学习下的训练深度学习模型工具。
- 按照以下说明设置参数选项卡下的变量:
- 对于输入训练数据,浏览至之前步骤中导出的训练数据的路径。
- 对于输出文件夹,浏览至您选择的计算机上的任一目录。
- 可以为最大轮数输入用于微调模型的迭代次数。
轮数是工具遍历数据所需进行的迭代次数。
- 对于预训练模型,输入从 ArcGIS Living Atlas of the World 下载的“北极海豹检测”(.dlpk) 文件。
- 对于批量大小,输入 4。 根据您的 GPU 能力增加或减少此数字。
批量大小应始终为平方数。
- 选中当模型停止改进时停止。
- 选中冻结模型。
- 设置环境选项卡下的变量。
- 对于处理器类型,根据需要选择 CPU 或 GPU。
如果可以使用 GPU,建议您选择 GPU 并设置待使用 GPU 的 GPU ID。
- 对于处理器类型,根据需要选择 CPU 或 GPU。
- 单击运行。
现在,您可以将此模型应用于影像数据,进行推理分析。