From 35f5c6d4e9eb7d2c1ad5fb4f5e3c520d1f0e8be4 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Tue, 2 Jul 2024 11:45:40 +0800 Subject: [PATCH] =?UTF-8?q?feat=20generate=20to=20product=20image=20?= =?UTF-8?q?=E6=96=B0=E5=A2=9E=20image=5Fstrength=E5=8F=82=E6=95=B0=20fix?= =?UTF-8?q?=20print=20=E9=80=8F=E6=98=8E=E5=9B=BE=E5=BC=82=E5=B8=B8?= =?UTF-8?q?=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/api/api_generate_image.py | 4 ++-- app/service/design/items/pipelines/painting.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py index 41cf989..cce1300 100644 --- a/app/api/api_generate_image.py +++ b/app/api/api_generate_image.py @@ -3,7 +3,7 @@ import logging from fastapi import APIRouter, BackgroundTasks, HTTPException -from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel +from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel, GenerateRelightImageModel from app.schemas.response_template import ResponseModel from app.service.generate_image.service_generate_image import GenerateImage, infer_cancel as generate_image_infer_cancel from app.service.generate_image.service_generate_product_image import GenerateProductImage, infer_cancel as generate_product_image_cancel @@ -145,7 +145,7 @@ def generate_product_image(tasks_id: str): @router.post("/generate_relight_image") -def generate_relight_image(request_item: GenerateProductImageModel, background_tasks: BackgroundTasks): +def generate_relight_image(request_item: GenerateRelightImageModel, background_tasks: BackgroundTasks): """ 创建一个具有以下参数的请求体: - **tasks_id**: 任务id 用于取消生成任务和获取生成结果 diff --git a/app/service/design/items/pipelines/painting.py b/app/service/design/items/pipelines/painting.py index a738455..424a395 100644 --- a/app/service/design/items/pipelines/painting.py +++ b/app/service/design/items/pipelines/painting.py @@ -118,6 +118,7 @@ class PrintPainting(object): print_background = cv2.cvtColor(np.array(source_image_pil), cv2.COLOR_RGBA2BGR) mask_background = cv2.cvtColor(np.array(source_image_pil_mask), cv2.COLOR_RGBA2BGR) + ret, mask_background = cv2.threshold(mask_background, 124, 255, cv2.THRESH_BINARY) else: mask = self.get_mask_inv(image) mask = np.expand_dims(mask, axis=2)