From 102dd53d68da96ab0915819c0f56630c6daa3699 Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Mon, 7 Oct 2024 10:24:01 +0800
Subject: [PATCH] =?UTF-8?q?feat=20fix=20=20=20=20=20=20sketch=20=E6=8F=90?=
 =?UTF-8?q?=E5=8F=96=E4=BF=AE=E5=A4=8D=E6=B2=A1=E6=9C=89=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E5=90=8E=E7=BC=80=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/service/lineart/service.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/app/service/lineart/service.py b/app/service/lineart/service.py
index 34f29dc..d822dfa 100644
--- a/app/service/lineart/service.py
+++ b/app/service/lineart/service.py
@@ -48,6 +48,11 @@ class LineArtService:
 
     def get_image(self):
         image = oss_get_image(bucket=self.image_url.split('/')[0], object_name=self.image_url[self.image_url.find('/') + 1:], data_type="cv2")
+        # 将其转换为彩色图像
+        if len(image.shape) == 3 and image.shape[2] == 4:
+            image = cv2.cvtColor(image, cv2.COLOR_BGRA2BGR)
+        elif len(image.shape) == 2:
+            image = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)
         return image
 
     def put_image(self, image):
@@ -84,10 +89,10 @@ class LineArtService:
 
 if __name__ == '__main__':
     request_item = Image2SketchModel(
-        image_url="aida-users/89/relight_image/d5f0d967-f8e8-424d-98f9-a8ad8313deec-0-89.png",
+        image_url="aida-collection-element/87/Sketchboard/555a443f-fd6b-4cd7-8147-b92d55513af0.png",
         default_style="4",
         sketch_bucket="test",
-        sketch_name="test123.jpg"
+        sketch_name="test123"
     )
     service = LineArtService(request_item)
     result_url = service.get_result()