From 20e5ead9c585d8266ec1adc270e768f2415117a0 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 25 Jul 2024 09:47:27 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20=20design=20pipeline=20=20contour?= =?UTF-8?q?=20detection=20=E6=96=B0=E5=A2=9E=E5=89=8D=E5=90=8E=E7=89=87?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design/items/pipelines/contour_detection.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/app/service/design/items/pipelines/contour_detection.py b/app/service/design/items/pipelines/contour_detection.py index 018dbca..487d2d6 100644 --- a/app/service/design/items/pipelines/contour_detection.py +++ b/app/service/design/items/pipelines/contour_detection.py @@ -43,7 +43,8 @@ class ContourDetection(object): result['mask'] = Mask else: result['mask'] = cv2.bitwise_and(Mask, result['pre_mask']) - + result['front_mask'] = result['mask'] + result['back_mask'] = result['mask'] return result @staticmethod