feat 新增对比度

app/service/generate_image/service.py

@@ -22,6 +22,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
 from app.service.generate_image.utils.upload_sd_image import upload_png_sd
 
@@ -84,6 +85,7 @@ class GenerateImage:
                 is_smudge, not_smudge_image = stain_detection(remove_bg_image)
                 image_result = not_smudge_image
             if is_smudge:  # 无污点
+                image_result = adjust_contrast(image_result)
                 image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png")
                 # logger.info(f"upload image SUCCESS ： {image_url}")
                 self.generate_data['status'] = "SUCCESS"