rv1126 移植yolov8 pose

airockchip / rknn_model_zoo

Apache License 2.0

1.05k stars 192 forks source link

rv1126 移植yolov8 pose #151

Open AI-LLM2 opened 3 months ago

AI-LLM2 commented 3 months ago

目前在移植yolov8 pose进入rv1126。转换的模型用python代码测试过是正常的。用c++代码推理结果整体score偏低只有0.1，框和关键点的位置也不对。我查了一下rk3588和rv1126的模型，发现输出的顺序反过来了。请问沿用3558 pose的代码修改后处理要怎么修改。rk3588的代码参考的是：https://github.com/cqu20160901/yolov8pose_rknn_Cplusplus ![Uploading image.png…]()

zen-xingle commented 3 months ago

可以参考这个https://github.com/airockchip/rknn_model_zoo/tree/main/examples/yolov8_pose，研发给的信息是下个版本会加上 rv1126/rv1109/rk1808 的支持