From 33e7d884cf200937ae2f64d0901824c8c097d7c7 Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Fri, 19 Jul 2024 15:17:54 +0800
Subject: [PATCH] =?UTF-8?q?feat=20=20=E6=B5=8B=E8=AF=95=E6=89=B9=E9=87=8F?=
 =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E5=9B=BE=E7=89=87=20fix?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/service/design/items/pipelines/keypoints.py    | 2 +-
 app/service/design/items/pipelines/loading.py      | 2 +-
 app/service/design/items/pipelines/painting.py     | 4 ++--
 app/service/design/items/pipelines/scale.py        | 2 +-
 app/service/design/items/pipelines/segmentation.py | 2 +-
 app/service/design/items/pipelines/split.py        | 2 +-
 app/service/design/service.py                      | 2 +-
 7 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/app/service/design/items/pipelines/keypoints.py b/app/service/design/items/pipelines/keypoints.py
index 1a264d6..fded7de 100644
--- a/app/service/design/items/pipelines/keypoints.py
+++ b/app/service/design/items/pipelines/keypoints.py
@@ -28,7 +28,7 @@ class KeypointDetection(object):
     # self.client.close()
     # print(f"client close time : {time.time() - start_time}")
 
-    @ ClassCallRunTime
+    # @ClassCallRunTime
     def __call__(self, result):
         # logging.info("KeypointDetection run ")
         if result['name'] in ['blouse', 'skirt', 'dress', 'outwear', 'trousers', 'tops', 'bottoms']:  # 查询是否有数据 且类别相同 相同则直接读 不同则推理后更新
diff --git a/app/service/design/items/pipelines/loading.py b/app/service/design/items/pipelines/loading.py
index ad5aec2..f0f4188 100644
--- a/app/service/design/items/pipelines/loading.py
+++ b/app/service/design/items/pipelines/loading.py
@@ -13,7 +13,7 @@ class LoadImageFromFile(object):
         self.print_dict = print_dict
         # self.minio_client = Minio(f"{MINIO_URL}", access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
 
-    @ClassCallRunTime
+    # @ClassCallRunTime
     def __call__(self, result):
         result['image'], result['pre_mask'] = self.read_image(self.path)
         result['gray'] = cv2.cvtColor(result['image'], cv2.COLOR_BGR2GRAY)
diff --git a/app/service/design/items/pipelines/painting.py b/app/service/design/items/pipelines/painting.py
index 8e4e524..07c6a84 100644
--- a/app/service/design/items/pipelines/painting.py
+++ b/app/service/design/items/pipelines/painting.py
@@ -14,7 +14,7 @@ class Painting(object):
     def __init__(self, painting_flag=True):
         self.painting_flag = painting_flag
 
-    @ClassCallRunTime
+    # @ClassCallRunTime
     def __call__(self, result):
         if result['name'] not in ['hairstyle', 'earring'] and self.painting_flag and result['color'] != 'none':
             dim_image_h, dim_image_w = result['image'].shape[0:2]
@@ -87,7 +87,7 @@ class PrintPainting(object):
     def __init__(self, print_flag=True):
         self.print_flag = print_flag
 
-    @ClassCallRunTime
+    # @ClassCallRunTime
     def __call__(self, result):
         single_print = result['print']['single']
         overall_print = result['print']['overall']
diff --git a/app/service/design/items/pipelines/scale.py b/app/service/design/items/pipelines/scale.py
index 43604cb..edd98c9 100644
--- a/app/service/design/items/pipelines/scale.py
+++ b/app/service/design/items/pipelines/scale.py
@@ -11,7 +11,7 @@ class Scaling(object):
     def __init__(self):
         pass
 
-    @ClassCallRunTime
+    # @ClassCallRunTime
     def __call__(self, result):
         if result['keypoint'] in ['waistband', 'shoulder', 'head_point']:
             # milvus_db_keypoint_cache
diff --git a/app/service/design/items/pipelines/segmentation.py b/app/service/design/items/pipelines/segmentation.py
index 4e6e0d0..7a1773c 100644
--- a/app/service/design/items/pipelines/segmentation.py
+++ b/app/service/design/items/pipelines/segmentation.py
@@ -15,7 +15,7 @@ class Segmentation(object):
         self.device = device
         self.debug = debug
 
-    @ClassCallRunTime
+    # @ClassCallRunTime
     def __call__(self, result):
         _, seg_result = self.load_seg_result(result["image_id"])
         if not _:
diff --git a/app/service/design/items/pipelines/split.py b/app/service/design/items/pipelines/split.py
index dd9becb..aab0f8d 100644
--- a/app/service/design/items/pipelines/split.py
+++ b/app/service/design/items/pipelines/split.py
@@ -18,7 +18,7 @@ class Split(object):
     Split image into front and back layer according to the segmentation result
     """
 
-    @ClassCallRunTime
+    # @ClassCallRunTime
     # KNet
     def __call__(self, result):
         try:
diff --git a/app/service/design/service.py b/app/service/design/service.py
index 3d71326..e5cfac7 100644
--- a/app/service/design/service.py
+++ b/app/service/design/service.py
@@ -160,7 +160,7 @@ def process_images(images):
     return results
 
 
-@RunTime
+# @RunTime
 def upload_images(image_obj):
     bucket_name = image_obj['image_url'].split("/", 1)[0]
     object_name = image_obj['image_url'].split("/", 1)[1]