From a0c47c336700496f3a85145928133f72c0bc0a76 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 25 Apr 2024 15:54:17 +0800 Subject: [PATCH] =?UTF-8?q?feat=20generate=20sketch=20=E6=B1=A1=E7=82=B9?= =?UTF-8?q?=E6=A3=80=E6=B5=8B=E6=94=B9=E4=B8=BA=E9=A1=B6=E9=83=A8=E4=B8=A4?= =?UTF-8?q?=E8=A7=92?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/generate_image/service.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py index 2cbd2dd..b78df3c 100644 --- a/app/service/generate_image/service.py +++ b/app/service/generate_image/service.py @@ -78,7 +78,10 @@ class GenerateImage: self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) else: image = result.as_numpy("generated_image") - image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_BGR2RGB) + if self.mode == "img2img": + image_result = np.squeeze(image.astype(np.uint8)) + else: + image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_BGR2RGB) is_smudge = True if self.category == "sketch": # 去背景