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