From 78d0eb73734238b24d32fd9adc2d920f8f8f231e Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Thu, 25 Apr 2024 14:11:09 +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 | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py
index 081d361..2cbd2dd 100644
--- a/app/service/generate_image/service.py
+++ b/app/service/generate_image/service.py
@@ -77,7 +77,8 @@ class GenerateImage:
             # self.generate_data['data'] = str(error)
             self.redis_client.set(self.tasks_id, json.dumps(self.generate_data))
         else:
-            image_result = result.as_numpy("generated_image")[0]
+            image = result.as_numpy("generated_image")
+            image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_BGR2RGB)
             is_smudge = True
             if self.category == "sketch":
                 # 去背景