search

airockchip / rknn_model_zoo

Apache License 2.0
1.03k stars 188 forks source link

yolov8-pose readme中的onnx和官方导出onnx检测头不一致 #170

Open pjdu opened 2 months ago

pjdu commented 2 months ago

按照model_zoo example readme导出的onnx检测头和ultralytics导出的检测头一致，但是和readme提供的已转换的onnx不一致，能否提供导出脚本？

CuberFan commented 2 months ago

按照model_zoo example readme导出的onnx检测头和ultralytics导出的检测头一致，但是和readme提供的已转换的onnx不一致，能否提供导出脚本？

很多算法都是经过rockchip官方修改的，用原版的没法适配NPU，使用rockchip修改版的yolov8-pose吧

pjdu commented 2 months ago

用rockchip修改版的yolov8-pose导出的onnx和model zoo提供的不一样

pubuzai commented 2 months ago

我参考https://www.youtube.com/watch?v=pN8mKZ5wpdQ里视频13钟分左右，成功导出和官方给的onnx一致的检测头的onnx,我用的是yolov8n的模型