移动端AI与ML应用开发：基于iOS和Android

Android

上使用

ML Kit

的计算机视觉应用程序

|

65

另一个常见选项是启用分类。由于默认的物体检测器只能检测 5 类物体，并且给了它们

一个非常通用的标签，因此我这里没有打开它，我们将使用本章前面讨论过的图像标记

API 来标记我们自己的图像。如果你想检测超过 5 个基本类的物体，则可以使用自定义

TensorFlow 模型来实现。我们将在第 9 ～ 11 章中探讨使用自定义模型。

4.2.5

第

5

步：处理按钮交互

当用户触摸按钮时，你需要调用物体检测器，获取其响应，然后从那里得到图像中物体

的边界框。稍后我们还会使用这些边界框将图像裁剪为边界框定义的子图像，以便我们

可以将其传递给标记器。但是现在，我们只实现物体检测处理程序。它应该是这样的：

btn.setOnClickListener {

val objectDetector = ObjectDetection.getClient(options)

var image = InputImage.fromBitmap(bitmap!!, 0)

objectDetector.process(image)

.addOnSuccessListener { detectedObjects ->

// Task completed successfully

}

.addOnFailureListener { e ->

// Task failed with an exception

// ...

}

}

因此，与你之前对图像标记所做的类似 ...