我试图在tensorflow中用本机模块(桥接)实现poseNet模型。我指的链接是https://github.com/tensorflow/examples/tree/master/lite/examples/posenet/android
在此,我只能找到单姿态估计的方法。在tflite版本的poseNet中,是否还有其他的多姿态估计参考?它是否支持tensorflow lite?
提前感谢
发布于 2022-01-04 13:15:45
首先,我要说,是的,多人姿态估计是由tflite支持的,但是对于它,您必须使用另一个tflite模型,而不是tensorflow posenet提出的模型,因为这个模型只支持一个人的检测。
我使用了颤振插件代码,用于在android (kotlin/java)和TFlite中对图像和视频进行多人检测。代码在我的GitHub存储库中,因为它是一个大代码,所以我不能在这里发布它。代码可以在这里找到(https://github.com/stevelaclasse/posenet/blob/main/app/src/main/java/cm/stevru/andropose/inference/DecodePose.java)。在这里,您将看到multiPoseEstimation()函数。
我还用一人模型试验了另一种多姿态估计方法。我使用了一个人员检测器模型来检测图像上的所有人员,然后我裁剪每个检测到的人,并创建一个新的调整大小的图像,与这个人,我发送给一个人的poseEstimation模型。它对好的图像效果很好,但是只有被检测到的人的裁剪图像失去了他的质量,这样poseEstimation就不会检测到图像上的所有姿势,而且在最后,对图像上的人估计的姿势很少。这就是我选择multiPerson poseEstimation模型的原因。
您也可以查看这个博客https://medium.com/flutter-community/posenet-for-ios-android-and-flutter-using-tensorflow-lite-836788a110c7,以更好地理解代码。
用于多人姿态估计的颤振插件代码可以在这里找到:https://github.com/shaqian/flutter_tflite/tree/master/android。
我希望它能对你或某人有帮助。
https://stackoverflow.com/questions/68170600
复制相似问题