diff --git a/csrcs/fastdeploy/vision/ppdet/ppyoloe.cc b/csrcs/fastdeploy/vision/ppdet/ppyoloe.cc index b2d3582c456..9c698976e0a 100644 --- a/csrcs/fastdeploy/vision/ppdet/ppyoloe.cc +++ b/csrcs/fastdeploy/vision/ppdet/ppyoloe.cc @@ -176,8 +176,7 @@ bool PPYOLOE::Postprocess(std::vector& infer_result, result->scores.push_back(nms.out_box_data[i * 6 + 1]); result->boxes.emplace_back(std::array{ nms.out_box_data[i * 6 + 2], nms.out_box_data[i * 6 + 3], - nms.out_box_data[i * 6 + 4] - nms.out_box_data[i * 6 + 2], - nms.out_box_data[i * 6 + 5] - nms.out_box_data[i * 6 + 3]}); + nms.out_box_data[i * 6 + 4], nms.out_box_data[i * 6 + 5]}); } } else { int box_num = 0; @@ -197,8 +196,7 @@ bool PPYOLOE::Postprocess(std::vector& infer_result, result->scores.push_back(box_data[i * 6 + 1]); result->boxes.emplace_back( std::array{box_data[i * 6 + 2], box_data[i * 6 + 3], - box_data[i * 6 + 4] - box_data[i * 6 + 2], - box_data[i * 6 + 5] - box_data[i * 6 + 3]}); + box_data[i * 6 + 4], box_data[i * 6 + 5]}); } } return true; diff --git a/model_zoo/vision/ppyoloe/ppyoloe.py b/model_zoo/vision/ppyoloe/ppyoloe.py index 7d79dfd8cf7..a3b12c1dc6b 100644 --- a/model_zoo/vision/ppyoloe/ppyoloe.py +++ b/model_zoo/vision/ppyoloe/ppyoloe.py @@ -14,7 +14,7 @@ # 预测图片 im = cv2.imread("000000014439_640x640.jpg") -result = model.predict(im, conf_threshold=0.5) +result = model.predict(im) # 可视化结果 fd.vision.visualize.vis_detection(im, result)