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

2024-04-25 12:47:45 +08:00
parent a955d3efe8
commit 6b9822b87f
1 changed files with 1 additions and 1 deletions
--- a/app/service/generate_image/service.py
+++ b/app/service/generate_image/service.py
@@ -85,7 +85,7 @@ class GenerateImage:
                # 污点检测
                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)
                image_result = not_smudge_image
            if is_smudge:  # 无污点