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