Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

请问pose的detection和landmark模型从tflite转换过来是参考哪个部分呢 #1

Open
sjr512 opened this issue Aug 16, 2022 · 8 comments

Comments

@sjr512
Copy link

sjr512 commented Aug 16, 2022

RT
跑了一下大佬的代码,发现pose的效果不是很好,想自己尝试转换一下官方的tflite模型到pytorch再试试,想问问大佬是怎么转的

@positive666
Copy link
Owner

RT 跑了一下大佬的代码,发现pose的效果不是很好,想自己尝试转换一下官方的tflite模型到pytorch再试试,想问问大佬是怎么转的

您好 你是指哪部分POSE,网上有脚本可以转的 我最近没在弄这个方法,因为想对接UE里面 Mediapipe可能因为精度和需要从新计算BS先搁置了

@sjr512
Copy link
Author

sjr512 commented Sep 6, 2022

RT 跑了一下大佬的代码,发现pose的效果不是很好,想自己尝试转换一下官方的tflite模型到pytorch再试试,想问问大佬是怎么转的

您好 你是指哪部分POSE,网上有脚本可以转的 我最近没在弄这个方法,因为想对接UE里面 Mediapipe可能因为精度和需要从新计算BS先搁置了

身体的部分(blazepose和blazepose_landmark),我仔细看了一下代码你项目里的模型是半身的,想问问大佬有全身的pytorch版本模型么

@positive666
Copy link
Owner

Author

** sjr512 ** commented 19分钟前

全身的也可以啊 而且你是想检测2D的还是3D的

@sjr512
Copy link
Author

sjr512 commented Sep 6, 2022

Author

** sjr512 ** commented 19分钟前

全身的也可以啊 而且你是想检测2D的还是3D的

用你项目里的pytorch模型全身是不行的吧?我把pose_connection的点补全了,模型输出是对不上的。我的话2D检测可以满足,有3D的更好。

@positive666
Copy link
Owner

Author

** sjr512 ** commented 19分钟前

全身的也可以啊 而且你是想检测2D的还是3D的

用你项目里的pytorch模型全身是不行的吧?我把pose_connection的点补全了,模型输出是对不上的。我的话2D检测可以满足,有3D的更好。

blazepose不是吗?你开启了吗 2D的话效果目前SOTA的是YOLOV5的POSE比较好 3D的话我用过英伟达的可以实时1080P

@sjr512
Copy link
Author

sjr512 commented Sep 6, 2022

Author

** sjr512 ** commented 19分钟前

全身的也可以啊 而且你是想检测2D的还是3D的

用你项目里的pytorch模型全身是不行的吧?我把pose_connection的点补全了,模型输出是对不上的。我的话2D检测可以满足,有3D的更好。

blazepose不是吗?你开启了吗 2D的话效果目前SOTA的是YOLOV5的POSE比较好 3D的话我用过英伟达的可以实时1080P

就是你项目里的blazepose.pth模型,是只能检测半身的,不是全身的。(开启是指--detect_pose?如果是这个那我肯定开启了,还是说有其他参数控制检测半身还是全身?)。那我去看看yolov5的pose检测,感谢大佬。

@positive666
Copy link
Owner

Author

** sjr512 ** commented 19分钟前

全身的也可以啊 而且你是想检测2D的还是3D的

用你项目里的pytorch模型全身是不行的吧?我把pose_connection的点补全了,模型输出是对不上的。我的话2D检测可以满足,有3D的更好。

blazepose不是吗?你开启了吗 2D的话效果目前SOTA的是YOLOV5的POSE比较好 3D的话我用过英伟达的可以实时1080P

就是你项目里的blazepose.pth模型,是只能检测半身的,不是全身的。(开启是指--detect_pose?如果是这个那我肯定开启了,还是说有其他参数控制检测半身还是全身?)。那我去看看yolov5的pose检测,感谢大佬。

可以替换成我仓库里的YOLO-POSE精度比这个高

@positive666
Copy link
Owner

mediapipe3d 版本升级了

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants