feat generate 新增色阶调整 亮度调整

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)
                 # 类型识别