在 Android 上使用搭载 TensorFlow 的 YOLO 网络进行实时物体检测
android-yolo 是 TensorFlow 在 Android 设备上的第一个 YOLO 实现。它与 Android Studio 兼容,开箱即用。它可以检测 Pascal VOC 数据集中的 20 种物体类别:飞机、自行车、鸟、船、瓶子、公共汽车、汽车、猫、椅子、牛、餐桌、狗、马、摩托车、人、盆栽、羊、沙发、火车和电视/显示器。目前,网络只会一次输出一个预测的边界框。未来,代码将会被扩展,以输出多个预测。
要使用此演示,首先克隆仓库。下载 TensorFlow YOLO 模型 并将其放在 android-yolo/app/src/main/assets 中。然后在 Android Studio 中打开项目。一旦项目打开,您可以在 Android 设备上运行项目,方法是使用"运行 'app'"命令并选择您的设备。
GPU 目前在 Android 上不受 TensorFlow 支持。如果您拥有相当强大的 Android 设备,则每秒可处理大约两帧图像。以下是一个 视频,展示了该应用的一个小示例。
暂无开放 Issues,或尚未同步最近议题。