创建 ML - 是否可以找到检测到的 object 的 position？

Question

我正在尝试使用 Create ML 构建 object 检测 model。 除了检测图片中物体的类型，我希望它在 output 中给我每个检测到的 object 的坐标（或位置）。

如何获取 output 图片中每个检测到的 object 的坐标？

是否可以使用 Create ML 做到这一点？ 如果是，如何？

Answer 1

是的，object 检测器返回它找到的VNRecognizedObjectObservation对象数组。 每个 object 都包含一个匹配标签数组、一个置信度值和一个边界框以及其他信息。

边界框是归一化的，所以在使用前需要用VNImageRectForNormalizedRect将其转换为像素坐标。

Apple 有一个出色的示例应用程序，其中包含详细说明，您可以在此处找到。