请问trt版本的yolo-pose是yolov5原版的预测还是带人体关键点的预测的?
Closed this issue · 9 comments
PaulX1029 commented
May I ask whether the yolo-pose of the trt version is the prediction of the original version of yolov5 or the prediction with the key points of the human body?
期待您的回复,谢谢!
fengxiuyaun commented
带pose的
fengxiuyaun commented
由于上传的匆忙,还没写详细的readme。大体上是先从pytorch通过gen_wts.py把模型导出来。然后使用yolo_pose里边的C代码生成.engine;然后可以使用对应的.py脚本调用engine了。
fengxiuyaun commented
欢迎star
PaulX1029 commented
已经star,我想借鉴您的c版本前后处理,这样是可以跑通的吧?
fengxiuyaun commented
应该是可以的(由于yolov5版本很多,网络结构上可能要稍微修改一下)。不过大多坑我都踩过了,有什么问题可以随时交流
PaulX1029 commented
好的,感谢!
PaulX1029 commented
我想再问一下 onnx推理的模型可以提供一下吗?用原版的官方的转换有点不成功
PaulX1029 commented
readme中给的模型下载链接都已经失效了(也可能是我这边打不开)
fengxiuyaun commented
Add WeChat ? My number is bphegyan2