修复 #2359 (#2363)

Update pptinypose_utils.cc 添加 py 范围限制, 避免 utils::DarkParse 内部出现数组越界
PaddlePaddle · Jan 29, 2024 · cfd80e9 · cfd80e9
1 parent cfebd24
commit cfd80e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastdeploy/vision/keypointdet/pptinypose/pptinypose_utils.cc b/fastdeploy/vision/keypointdet/pptinypose/pptinypose_utils.cc
@@ -101,7 +101,7 @@ void GetFinalPredictions(const std::vector<float>& heatmap,
     coords[j * 2 + 1] = idx / heatmap_width;
     int px = int(coords[j * 2] + 0.5);
     int py = int(coords[j * 2 + 1] + 0.5);
-    if (DARK && px > 1 && px < heatmap_width - 2) {
+    if (DARK && px > 1 && px < heatmap_width - 2 && py > 1 && py < heatmap_height - 2) {
       utils::DarkParse(heatmap, dim, &coords, px, py, index, j);
     } else {
       if (px > 0 && px < heatmap_width - 1) {