aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat

Browse Source 
fix      sketch 提取修复没有文件后缀问题
This commit is contained in:
zhouchengrong
2024-10-07 10:24:01 +08:00
parent 4e420f8ae8
commit 102dd53d68
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/service/lineart/service.py
View File

@@ -48,6 +48,11 @@ class LineArtService:
def get_image(self): 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") 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 return image
def put_image(self, image): def put_image(self, image):
@@ -84,10 +89,10 @@ class LineArtService:
if __name__ == '__main__': if __name__ == '__main__':
request_item = Image2SketchModel( 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", default_style="4",
sketch_bucket="test", sketch_bucket="test",
sketch_name="test123.jpg" sketch_name="test123"
) )
service = LineArtService(request_item) service = LineArtService(request_item)
result_url = service.get_result() result_url = service.get_result()