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

feat generate sketch 污点检测改为顶部两角

Browse Source
This commit is contained in:
zhouchengrong
2024-04-25 12:47:45 +08:00
parent a955d3efe8
commit 6b9822b87f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -85,7 +85,7 @@ class GenerateImage:
# 污点检测 # 污点检测
is_smudge, not_smudge_image = stain_detection(remove_bg_image) is_smudge, not_smudge_image = stain_detection(remove_bg_image)
# 类型识别 # 类型识别
category, scores, not_smudge_image = generate_category_recognition(image=image_result, gender=self.gender) category, scores, not_smudge_image = generate_category_recognition(image=remove_bg_image, gender=self.gender)
self.generate_data['category'] = str(category) self.generate_data['category'] = str(category)
image_result = not_smudge_image image_result = not_smudge_image
if is_smudge: # 无污点 if is_smudge: # 无污点