How to use outputs of posenet model in tflite

前端 未结 1 1370
遇见更好的自我
遇见更好的自我 2021-01-01 08:06

I am using the tflite model for posenet from here. It takes input 1*353*257*3 input image and returns 4 arrays of dimens 1*23*17*17, 1*23*17*34, 1*23*17*64 and 1*23*17*1. Th

相关标签:
1条回答
  • 2021-01-01 08:20

    I think there are something wrong with this tflite model file. So I tried to create the posenet tflite model using the weights in the model. All the weights in the model can be downloaded from tfjs-models: https://github.com/tensorflow/tfjs-models/tree/master/posenet

    Then you can generate the model and do all the pre and post process as the follow repo: https://github.com/zg9uagfv/tf_posenet

    After the posenet model generated, you can export to .pb file or .tflite file. I have tried the process successfully, and the posenet model can be run in my Android App with GPU successfully.

    0 讨论(0)
提交回复
热议问题