From 48cfb4c1caa69ae2100d956fbd25e7fb56d323ef Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Fri, 10 Jan 2025 15:07:06 +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=97=AD=E8=B0=83=E8=AF=95=E6=97=A5=E5=BF=97=E6=89=93?=
 =?UTF-8?q?=E5=8D=B0=20docs=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4?=
 =?UTF-8?q?=EF=BC=89:=20refactor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20te?=
 =?UTF-8?q?st(=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_batch/pipeline/segmentation.py | 2 +-
 app/service/design_fast/design_generate.py        | 2 +-
 app/service/design_fast/pipeline/segmentation.py  | 2 +-
 app/service/utils/decorator.py                    | 3 ++-
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/app/service/design_batch/pipeline/segmentation.py b/app/service/design_batch/pipeline/segmentation.py
index 8ebcd3a..ca7da1c 100644
--- a/app/service/design_batch/pipeline/segmentation.py
+++ b/app/service/design_batch/pipeline/segmentation.py
@@ -58,7 +58,7 @@ class Segmentation:
     @staticmethod
     def load_seg_result(image_id):
         file_path = f"seg_cache/{image_id}.npy"
-        logger.info(f"load seg file name is :{SEG_CACHE_PATH}{image_id}.npy")
+        # logger.info(f"load seg file name is :{SEG_CACHE_PATH}{image_id}.npy")
         try:
             seg_result = np.load(file_path)
             return True, seg_result
diff --git a/app/service/design_fast/design_generate.py b/app/service/design_fast/design_generate.py
index 4fc0726..2f7fa93 100644
--- a/app/service/design_fast/design_generate.py
+++ b/app/service/design_fast/design_generate.py
@@ -207,7 +207,7 @@ def design_generate_v2(request_data):
             'Connection': "keep-alive",
             'Content-Type': "application/json"
         }
-        logger.info(items_response)
+        # logger.info(items_response)
         response = post_request(url, json_data=items_response, headers=headers)
         if response:
             # 打印结果
diff --git a/app/service/design_fast/pipeline/segmentation.py b/app/service/design_fast/pipeline/segmentation.py
index 5e392ed..4828b33 100644
--- a/app/service/design_fast/pipeline/segmentation.py
+++ b/app/service/design_fast/pipeline/segmentation.py
@@ -73,7 +73,7 @@ class Segmentation:
     @staticmethod
     def load_seg_result(image_id):
         file_path = f"{SEG_CACHE_PATH}{image_id}.npy"
-        logger.info(f"load seg file name is :{SEG_CACHE_PATH}{image_id}.npy")
+        # logger.info(f"load seg file name is :{SEG_CACHE_PATH}{image_id}.npy")
         try:
             seg_result = np.load(file_path)
             return True, seg_result
diff --git a/app/service/utils/decorator.py b/app/service/utils/decorator.py
index bc7e9f1..c0164ab 100644
--- a/app/service/utils/decorator.py
+++ b/app/service/utils/decorator.py
@@ -22,7 +22,8 @@ def ClassCallRunTime(func):
         end_time = time.time()
         execution_time = end_time - start_time
         class_name = args[0].__class__.__name__  # 获取类名
-        print(f"class name: {class_name} , run time is : {execution_time} s")
+        if execution_time > 0.05:
+            logging.info(f"class name: {class_name} , run time is : {execution_time} s")
         return result
 
     return wrapper