search

fire717 / movenet.pytorch

A Pytorch implementation of MoveNet from Google. Include training code and pre-trained model.
MIT License
374 stars 87 forks source link

请问作者，如何对自己选定的图片或者视频进行姿态估计处理。 #18

Open ax642511 opened 2 years ago

ax642511 commented 2 years ago

作者您好，我使用您的模型进行了训练，最后生成了热力图和关键点检测，但是没有找到对测试图片进行姿态估计效果图检测的文件，请指点一下！万分感谢

fire717 commented 2 years ago

“最后生成了热力图和关键点检测”是什么意思？你不是都生成了吗？ 还是说生成了模型？ 训练好得到模型，可以参考readme的第五步和第六步：

# 5.After training finished, you need to change the test model path to test. Such as this in predict.py
run_task.modelLoad("output/xxx.pth")
# 6.run predict to show predict result, or run evaluate.py to compute my acc on test dataset.
python predict.py
ax642511 commented 2 years ago

是的，热力图和关键点检测都生成了，请问您的源码里有将这些关键点连接起来的py文件，即完整的骨骼图

fire717 commented 2 years ago

这个貌似没有，你根据需要按顺序绘制cv2.line即可，很简单