如何在Flutter上设置tflite PoseNet“单人姿势估计”?

时间:2020-08-04 00:33:19

标签: flutter dart tf-lite

我在一个使用tflite Posenet在Flutter Framework的移动设备上运行的项目中。我们希望在测试中获得更高的精度得分,但我们意识到Dart API文档https://pub.dev/documentation/tflite/latest/上原始示例提供的存储库使用了Defalt的多人位姿估计。 (提到的存储库:https://github.com/shaqian/flutter_realtime_detection)。

我们知道原始的tfjs存储库对此进行了讨论,并给出了示例https://github.com/tensorflow/tfjs-models/tree/master/posenet,但是我们在Dart API上找不到它。

如何设置单人姿势估计?

2 个答案:

答案 0 :(得分:2)

还有一个tflite_flutter插件正在开发中:https://pub.dev/packages/tflite_flutter。请看看。

此插件允许您运行任何自定义的.tflite模型,因此您应该能够从官方的TFLite网站(https://www.tensorflow.org/lite/models/pose_estimation/overview)下载单人posenet模型。

没有正式的flutter示例,但是您应该可以参考Android / iOS示例以了解如何对数据进行预处理/后处理。

答案 1 :(得分:2)

嗨,大家好,更新​​问题:

在使用Tflite.runPoseNetOnImage函数进行一些测试之后,我们对其进行了增强。该函数具有一个名为numResults的属性,我们可以将其设置为“ 1”。 Tflite.runPoseNetOnFrame具有相同的属性。

您可以在此链接(https://pub.dev/documentation/tflite/latest/)中看到它