From f21b7203bb1a7e7c9a6ca2280413a75f3dd8b756 Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Mon, 20 Jan 2025 11:26:42 +0800
Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=EF=BC=89:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20?=
 =?UTF-8?q?=E5=85=B3=E9=94=AE=E7=82=B9=E6=A8=A1=E5=9E=8B=E9=A2=84=E5=A4=84?=
 =?UTF-8?q?=E7=90=86=E5=90=8E=E5=A4=84=E7=90=86=E5=A2=9E=E5=8A=A0=E7=99=BD?=
 =?UTF-8?q?=E8=BE=B9=E6=A1=86=20docs=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98?=
 =?UTF-8?q?=E6=9B=B4=EF=BC=89:=20refactor=EF=BC=88=E9=87=8D=E6=9E=84?=
 =?UTF-8?q?=EF=BC=89:=20test(=E5=A2=9E=E5=8A=A0=E6=B5=8B=E8=AF=95):?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/service/design_fast/utils/design_ensemble.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/service/design_fast/utils/design_ensemble.py b/app/service/design_fast/utils/design_ensemble.py
index bfc50c6..8eef4f2 100644
--- a/app/service/design_fast/utils/design_ensemble.py
+++ b/app/service/design_fast/utils/design_ensemble.py
@@ -25,6 +25,7 @@ from app.core.config import *
 
 def keypoint_preprocess(img_path):
     img = mmcv.imread(img_path)
+    img = cv2.copyMakeBorder(img, 25, 25, 25, 25, cv2.BORDER_CONSTANT, value=[255, 255, 255])
     img_scale = (256, 256)
     h, w = img.shape[:2]
     img = cv2.resize(img, img_scale)
@@ -62,7 +63,11 @@ def keypoint_postprocess(output, scale_factor):
     scale_matrix = np.diag(scale_factor)
     nan = np.isinf(scale_matrix)
     scale_matrix[nan] = 0
-    return np.ceil(np.dot(segment_result, scale_matrix) * 4)
+    # 应用缩放因子
+    scaled_result = np.ceil(np.dot(segment_result, scale_matrix) * 4)
+    # 补偿边框偏移
+    compensated_result = scaled_result - 25
+    return compensated_result
 
 
 """