From 8bdcd15723d735de8a4f36816d4a94fc27003673 Mon Sep 17 00:00:00 2001
From: ShawnXsw <33308054+ShawnXsw@users.noreply.github.com>
Date: Sun, 24 Apr 2022 07:54:05 -0500
Subject: [PATCH] fix dark postprocess in cpp infer deploy( #5751) (#5815)

---
 deploy/cpp/src/keypoint_postprocess.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deploy/cpp/src/keypoint_postprocess.cc b/deploy/cpp/src/keypoint_postprocess.cc
index 52ac8d3d3..405195c3e 100644
--- a/deploy/cpp/src/keypoint_postprocess.cc
+++ b/deploy/cpp/src/keypoint_postprocess.cc
@@ -193,7 +193,7 @@ void get_final_preds(std::vector<float>& heatmap,
     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){
       dark_parse(heatmap, dim, coords, px, py, index, j);
     }
     else{
-- 
GitLab