标注 | 说明 | 数据类型 |
输入训练数据 | 包含训练模型所需的影像片、标注和统计数据的文件夹。 此数据为导出训练数据进行深度学习工具的输出。 当满足以下条件时,支持多个输入文件夹:
| Folder |
输出文件夹 | 将存储训练模型的输出文件夹位置。 | Folder |
最大轮数 (可选) | 将用于训练模型的最大轮数。 最大轮数值为 1 意味着数据集将通过神经网络向前和向后传递一次。 默认值为 20。 | Long |
模型类型 (可选) | 指定将用于训练深度学习模型的模型类型。
| String |
批大小 (可选) | 一次需要处理以便用于训练的训练样本数。 增加批处理大小可以提高工具性能;但是,随着批处理大小的增加,会占用更多内存。 如果可用于批量大小集的 GPU 内存不足,则该工具将尝试估计并使用最佳批量大小。 如果发生内存不足错误,请使用较小的批处理大小。 | Long |
模型参数 (可选) | 将使用模型类型参数中的信息填充此参数。 这些参数将有所不同,具体取决于模型架构。 下面介绍了在 ArcGIS 中训练的模型支持的模型参数。 ArcGIS 预训练模型和自定义深度学习模型可能具有该工具支持的其他参数。 有关适用于每种模型类型的参数的详细信息,请参阅深度学习参数。
| Value Table |
学习率 (可选) | 在整个训练过程中,现有信息将被新获取的信息覆盖的比率。 如果未指定任何值,则系统将在训练过程中从学习曲线中提取最佳学习率。 | Double |
骨干模型 (可选) | 指定要用作训练新模型的架构的、预先配置的神经网络。 这种方法称为迁移学习。 此外,从 PyTorch Image Models (timm) 指定受支持卷积神经网络时还可以使用 timm 作为前缀,例如 timm:resnet31、timm:inception_v4、timm:efficientnet_b3 等。
| String |
预训练模型 (可选) | 将用于微调新模型的预训练模型。 输入为 Esri 模型定义文件 (.emd) 或深度学习包文件 (.dlpk)。 可以对具有相似类的预训练模型进行微调以适应新模型。 预训练模型必须已使用将用于训练新模型的相同模型类型和骨干模型进行了训练。 | File |
验证百分比 (可选) | 将用于验证模型的训练样本的百分比。 默认值为 10。 | Double |
当模型停止改进时停止 (可选) | 指定是否将实施提前停止。
| Boolean |
冻结模型 (可选) | 指定是否冻结预训练模型中的骨干层,以使权重和偏差保持原始设计。
| Boolean |
数据增强 (可选) | 指定要使用的数据增强类型。 数据增强方法使用现有数据创建经过修改的数据集副本,实现人为增加训练集合。
| String |
增强参数 (可选) | 在增强参数中指定每个转换对应的值。
| Value Table |
影像片大小 (可选) | 用于训练模型的影像大小。 影像将裁剪为指定的影像片大小。 如果影像大小小于参数值,则将使用影像大小。 默认值为 224 像素。 | Long |
将大小调整为 (可选) | 调整影像片大小。 在调整影像片大小后,将裁剪影像片大小的像素块,以用于训练。 该参数将仅应用于对象检测 (PASCAL VOC)、对象分类(已标注切片)和超分辨率数据。 大小调整值通常为影像片大小值的一半。 如果此大小调整值小于影像片大小值,则将使用大小调整值创建像素块进行训练。 | String |
权重初始化方案 (可选) | 指定要用于为图层初始化权重的方案。 要使用多光谱数据训练模型,模型必须容纳不同类型的可用波段。 为此,需要重新初始化模型中的第一个图层。 仅当在模型中使用多光谱影像时,此参数才适用。
| String |
监控指标 (可选) | 指定在检查点设置和提前停止时要监控的指标。
| String |
派生输出
标注 | 说明 | 数据类型 |
输出模型 | 输出已训练模型文件。 | File |