From 16c05c0f5b36cf0096805b60e5caf12166ba1635 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 22 Jul 2024 15:41:07 +0800 Subject: [PATCH] =?UTF-8?q?feat=20=20=E5=A4=84=E7=90=86=E5=A4=9A=E5=B1=82s?= =?UTF-8?q?ketch=20design=E5=87=BA=E7=8E=B0=E7=9A=84=E5=86=85=E6=90=AD?= =?UTF-8?q?=E8=B6=8A=E7=95=8C=E9=97=AE=E9=A2=98=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design/items/pipelines/segmentation.py | 2 +- app/service/design/items/pipelines/split.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/app/service/design/items/pipelines/segmentation.py b/app/service/design/items/pipelines/segmentation.py index 7a1773c..8782e75 100644 --- a/app/service/design/items/pipelines/segmentation.py +++ b/app/service/design/items/pipelines/segmentation.py @@ -3,7 +3,6 @@ import os import numpy as np from app.core.config import SEG_CACHE_PATH -from app.service.utils.decorator import ClassCallRunTime from ..builder import PIPELINES from ...utils.design_ensemble import get_seg_result @@ -18,6 +17,7 @@ class Segmentation(object): # @ClassCallRunTime def __call__(self, result): _, seg_result = self.load_seg_result(result["image_id"]) + result['seg_result'] = seg_result if not _: result['seg_result'] = get_seg_result(result["image_id"], result['image']) self.save_seg_result(result['seg_result'][0], result['image_id']) diff --git a/app/service/design/items/pipelines/split.py b/app/service/design/items/pipelines/split.py index 031d644..f3da4e7 100644 --- a/app/service/design/items/pipelines/split.py +++ b/app/service/design/items/pipelines/split.py @@ -28,9 +28,9 @@ class Split(object): result['front_mask'] = result['mask'].copy() result['back_mask'] = np.zeros_like(result['mask']) else: - temp_front = result['seg_result'] == 1 + temp_front = result['seg_result'] == 1.0 result['front_mask'] = (result['mask'] * (temp_front + 0).astype(np.uint8)) - temp_back = result['seg_result'] == 2 + temp_back = result['seg_result'] == 2.0 result['back_mask'] = (result['mask'] * (temp_back + 0).astype(np.uint8)) if result['name'] in ('outwear', 'dress', 'blouse', 'skirt', 'trousers', 'tops', 'bottoms'):