yolov8_pose 全身姿态估计 pytorch转onnx转昇腾部署
收藏回复举报
yolov8_pose 全身姿态估计 pytorch转onnx转昇腾部署
新人帖
发表于2024-03-11 01:20:48
0 查看

模型代码都在附件,10mb限制上传不了太大的,需要高精度模型可以去yolov8的github下载,也上传不了py文件,把.txt改成.py就行

优点是推理速度非常快,一张图仅需5.4ms

atc命令,onnx->om:

atc --model=v8_n_pose_sim.onnx --framework=5 --output=v8_n_pose_sim --input_format=NCHW --input_shape="images:1,3,512,512" --log=error --soc_version=Ascend310B4

其中--model,--output,--input_shape,--soc_version改成自己的

本文是视频推理,如果需要改成实时推理只需要将

cv2.VideoCapture('./test_video/input_video.mp4')

改成

cv2.VideoCapture(0)

就行了

结果:

5_Y35MXT8N5Z7YC%5D2_HP(%5DF.png

本帖最后由 匿名用户2024/03/11 14:44:56 编辑

我要发帖子