From 6b9822b87f7639468c04dda612d9b729d9c6520c Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 25 Apr 2024 12:47:45 +0800 Subject: [PATCH] =?UTF-8?q?feat=20generate=20sketch=20=E6=B1=A1=E7=82=B9?= =?UTF-8?q?=E6=A3=80=E6=B5=8B=E6=94=B9=E4=B8=BA=E9=A1=B6=E9=83=A8=E4=B8=A4?= =?UTF-8?q?=E8=A7=92?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/generate_image/service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py index 0292223..081d361 100644 --- 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: # 无污点