diff --git a/app/service/generate_image/service_generate_image.py b/app/service/generate_image/service_generate_image.py index b49433e..daad5c6 100644 --- a/app/service/generate_image/service_generate_image.py +++ b/app/service/generate_image/service_generate_image.py @@ -91,7 +91,7 @@ class GenerateImage: image = result.as_numpy("generated_image") image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_RGB2BGR) is_smudge = True - if self.category == "sketch": + if self.category == "sketch" and self.version == "fast": # 色阶调整 cutoff = 1 levels_img = autoLevels(image_result, cutoff)