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

feat generate 污点图保存

Browse Source
This commit is contained in:
zhouchengrong
2024-05-18 08:21:50 +08:00
parent a23eca7850
commit 772cb12ced
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -93,9 +93,9 @@ class GenerateImage:
# 去背景 # 去背景
remove_bg_image = remove_background(luminance) remove_bg_image = remove_background(luminance)
# 人脸检测 # 人脸检测
if face_detect_pic(remove_bg_image, self.user_id, self.category, self.tasks_id) > 0: # if face_detect_pic(remove_bg_image, self.user_id, self.category, self.tasks_id) > 0:
is_smudge = False # is_smudge = False
else: # else:
# 污点/ # 污点/
is_smudge, not_smudge_image = stain_detection(remove_bg_image, self.user_id, self.category, self.tasks_id) is_smudge, not_smudge_image = stain_detection(remove_bg_image, self.user_id, self.category, self.tasks_id)
# 类型识别 # 类型识别