From e021acb190a1e5db20489429361e54158bada4a2 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 25 Apr 2024 17:56:25 +0800 Subject: [PATCH] =?UTF-8?q?feat=20generate=20=E6=96=B0=E5=A2=9E=E8=89=B2?= =?UTF-8?q?=E9=98=B6=E8=B0=83=E6=95=B4=20=E4=BA=AE=E5=BA=A6=E8=B0=83?= =?UTF-8?q?=E6=95=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/generate_image/service.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py index d1158ff..cb12fa5 100644 --- a/app/service/generate_image/service.py +++ b/app/service/generate_image/service.py @@ -23,7 +23,7 @@ from tritonclient.utils import np_to_triton_dtype from app.core.config import * from app.schemas.generate_image import GenerateImageModel from app.service.generate_image.utils.adjust_contrast import adjust_contrast -from app.service.generate_image.utils.image_processing import remove_background, stain_detection, generate_category_recognition +from app.service.generate_image.utils.image_processing import remove_background, stain_detection, generate_category_recognition, autoLevels, luminance_adjust from app.service.generate_image.utils.upload_sd_image import upload_png_sd logger = logging.getLogger() @@ -85,8 +85,13 @@ class GenerateImage: image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_RGB2BGR) is_smudge = True if self.category == "sketch": + # 色阶调整 + cutoff = 1 + levels_img = autoLevels(image_result, cutoff) + # 亮度调整 + luminance = luminance_adjust(0.3, levels_img) # 去背景 - remove_bg_image = remove_background(np.asarray(image_result)) + remove_bg_image = remove_background(luminance) # 污点检测 is_smudge, not_smudge_image = stain_detection(remove_bg_image) # 类型识别