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