From b749ba71b59132194e460d82abf5d9501b81fdf3 Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Thu, 25 Apr 2024 13:38:41 +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/utils/image_processing.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/app/service/generate_image/utils/image_processing.py b/app/service/generate_image/utils/image_processing.py
index ed9fb59..303fc52 100644
--- a/app/service/generate_image/utils/image_processing.py
+++ b/app/service/generate_image/utils/image_processing.py
@@ -205,3 +205,8 @@ def generate_category_recognition(image, gender):
         else:
             category = 'Outwear'
     return category, scores, image
+
+
+if __name__ == '__main__':
+    image = cv2.imread("1.png")
+    remove_background(image)