search

positive666 / mediapipe_PoseEstimation_pytorch

based on mediapipe human Estimation project for pytorch, provide function(face_mesh,iris,pose,hand)
Apache License 2.0
64 stars 10 forks source link

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

Open sjr512 opened 2 years ago

sjr512 commented 2 years ago

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

positive666 commented 2 years ago

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

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

sjr512 commented 2 years ago

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

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

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

positive666 commented 2 years ago

Author

sjr512 commented 19分钟前

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

sjr512 commented 2 years ago

Author

sjr512 commented 19分钟前

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

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

positive666 commented 2 years ago

Author

sjr512 commented 19分钟前

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

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

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

sjr512 commented 2 years ago

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 commented 1 year ago

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 commented 1 year ago

mediapipe3d 版本升级了