Skip To Content

使用模型

您可以在 ArcGIS Pro 中的 Image Analyst 工具箱提供的使用深度学习检测对象工具中使用此模型。 要使用该模型检测图像中各种对象的掩码,请完成以下步骤:

检测对象

要检测影像中各种对象的掩码,请完成以下步骤:

  1. 确保您已下载 Segment Anything Model (SAM) 预训练模型并且已在 ArcGIS Pro 中添加影像图层。
  2. 缩放至感兴趣区域或者使用整个图像。
    感兴趣区域
  3. 单击分析选项卡,然后浏览至工具
    ArcGIS Pro 中“分析”选项卡上的“工具”
  4. 单击地理处理窗格中的工具箱选项卡,选择 Image Analyst 工具,然后浏览至深度学习下的使用深度学习检测对象工具。
    “使用深度学习检测对象”工具
  5. 按照以下说明设置参数选项卡上的变量:
    1. 输入栅格 - 选择影像。
    2. 输出检测对象 - 设置输出要素类,其中将包含检测到的对象。
    3. 模型定义 - 选择预训练模型或微调模型的 .dlpk 文件。
    4. 参数 - 根据需要更改参数值。
      • padding - 影像切片边界处的像素数,将根据这些像素为相邻切片混合预测。 增加此值以减少边伪影,使输出更加平滑。 内边距的最大值可以是切片大小值的一半。
      • batch_size - 模型推断每个步骤中处理的影像切片数。 这取决于显卡的内存。
      • box_nms_thresh - 非极大值抑制用于过滤重复掩码的框 IoU 中断。
      • points_per_batch - 设置模型同时运行的点数。 数值越大,则速度可能越快,但将使用更多 GPU 内存。
      • stability_score_thresh - 置信度得分高于此阈值的检测将包含在结果中。 允许的值范围为从 0 到 1.0。
      • min_mask_region_area - 如果 >0,则将应用后处理以移除面积小于 min_mask_region_area 的掩码中的断开连接区域和孔洞。
    5. 非极大值抑制 - 可以选择性地选中复选框以移除具有较低置信度的重叠要素。

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

      • 设置置信度得分字段
      • 设置类值字段(可选)。
      • 设置最大重叠比(可选)。
    “参数”选项卡
  6. 环境选项卡上设置变量,选择 CPUGPU 作为处理器类型

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

    “环境”选项卡
  7. 单击运行

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

    输出图层

在本主题中
  1. 检测对象