瀏覽代碼

pose bug fix

leon 3 周之前
父節點
當前提交
c43aa1c5f1
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/infer/trt/yolo/yolo.cu

+ 1 - 1
src/infer/trt/yolo/yolo.cu

@@ -764,7 +764,7 @@ Result YoloModelImpl::forwards(void *stream)
                 result_object_box.keypoints.reserve(KEY_POINT_NUM);
                 for (int i = 0; i < KEY_POINT_NUM; i++)
                 {
-                    result_object_box.keypoints.emplace_back(pbox[8+i*3], pbox[8+i*3+1], pbox[8+i*3+2]);
+                    result_object_box.keypoints.emplace_back(pbox[9+i*3], pbox[9+i*3+1], pbox[9+i*3+2]);
                 }
             }
             else if (model_type_ == ModelType::YOLO11SEG)