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

feat 新增 生成sketch时对图片清理背景，剔除带有污点的结果图

Browse Source
This commit is contained in:
zchen
2024-04-23 15:43:03 +08:00
parent 51540d2970
commit 2c2b0d3b0d
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/service/generate_image/service.py
View File

@@ -84,6 +84,7 @@ class GenerateImage:
if is_smudge: # 无污点 if is_smudge: # 无污点
image_result = not_smudge_image image_result = not_smudge_image
image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png") image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png")
logger.info(f"upload image SUCCESS ： {image_url}")
self.generate_data['status'] = "SUCCESS" self.generate_data['status'] = "SUCCESS"
self.generate_data['message'] = "success" self.generate_data['message'] = "success"
self.generate_data['data'] = str(image_url) self.generate_data['data'] = str(image_url)