feat

fix      sketch 提取修复没有文件后缀问题

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()