移动设备上的视频分割

Question

我使用 tensorflow.js 在网络浏览器中对实时视频进行人物分割。是否有类似的 tensorflowlite/移动版本？是否有任何 android SDK 可用于在移动客户端上进行人/视频分割？任何指针都会非常有帮助。

谢谢 苏米亚

Answer 1

是的，TensorFlow Lite 可用于 Android。 Click here for more.

Answer 2

您可以使用TensorFlow Lite's Segmentation Model。您还可以在页面上看到一个示例 GIF。分割模型（DeepLab v3）可以分割图像中的各种对象。由于我们关注人物分割，DeepLabv3 也可以处理。

如果您想使用BodyPix，这是一个 TensorFlow.js 实现，您需要找到模型的冻结图以及输入/输出节点。只有这样，您才能将其放入 TensorFlow Lite。在调查 repo 时，我发现了这个 file，其中包含模型的 URL。

否则，我们可以使用Real-Time Pose Estimation，这也是一个TF JS实现。