Skip To Content

使用模型

可以在 ArcGIS Pro 的 Image Analyst 工具箱中提供的使用深度学习检测对象工具中使用传输 H 型结构检测模型。

要使用“传输 H 型结构检测”预训练模型,请完成以下步骤:

  1. 下载模型,然后在 ArcGIS Pro 中添加影像图层。

    添加数据。

  2. 单击分析选项卡,然后浏览至工具
    ArcGIS Pro 中“分析”选项卡上的“工具”
  3. 地理处理窗格中,单击工具箱,然后展开 Image Analyst 工具。 选择深度学习下的使用深度学习检测对象工具。
    “使用深度学习检测对象”工具
  4. 参数选项卡上,按照以下说明设置变量:
    1. 输入栅格 - 选择图像。
    2. 输出检测对象 - 设置输出要素类,其中将包含检测到的对象。
    3. 模型定义 - 选择预训练模型或微调模型的 .dlpk 文件。
    4. 参数(可选)- 根据需要更改参数值。
      • threshold - 置信度得分高于此阈值的检测将包含在结果中。 允许的值范围为从 0 到 1.0。
      • nms_overlap - 边界框之间的重叠比率,用于过滤置信度得分较低的检测。
      • test_time_augmentation - 在预测期间执行测试时数据增强。 这是一种用于提高模型预测的稳健性和准确性的技术。 它在推断过程中应用数据增强技术,这意味着生成测试数据的多个稍作修改的版本并聚合预测。 如果值为 True,则输入影像的翻转和旋转方向的预测将合并到最终输出,并且其置信度将取平均值。 对于仅在影像的某些方向上检测到的对象,这可能导致置信度低于阈值。
    5. 非极大值抑制 - 可以选择性地选中复选框以移除具有较低置信度的重叠要素。

      如果选中了此复选框,请执行以下操作:

      • 置信度得分字段 - 使用默认值。
      • 类值字段 - 使用默认值。
      • 最大重叠比 - 将最大重叠比值设置为 0.25。
    “使用深度学习检测对象”工具
  5. 环境选项卡上,按照以下说明设置变量:
    1. 处理范围 - 从下拉菜单中选择默认或任意其他选项。
    2. 处理器类型 - 根据需要选择 CPUGPU

      如果可以使用 GPU,建议您选择 GPU 并设置待使用 GPU 的 GPU ID

    “使用深度学习检测对象”工具环境
  6. 单击运行

    处理完成后,输出图层将添加到地图。

    检测结果