From 63f0585ce86f0d93152746a52394f4bdd998b2d9 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Wed, 12 Feb 2025 17:39:36 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=89:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20=20p?= =?UTF-8?q?roduct=20generate=20=E8=8E=B7=E5=8F=96=E4=B8=89=E9=80=9A?= =?UTF-8?q?=E9=81=93RGB=E5=9B=BE=E7=89=87=E6=97=B6=20=E6=97=A0=E6=B3=95?= =?UTF-8?q?=E8=8E=B7=E5=8F=96A=E9=80=9A=E9=81=93mask=20=E5=A2=9E=E5=BC=BA?= =?UTF-8?q?=E5=9B=BE=E7=89=87mode=E5=88=A4=E6=96=AD=E9=80=BB=E8=BE=91=20do?= =?UTF-8?q?cs=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20ref?= =?UTF-8?q?actor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E?= =?UTF-8?q?=E5=8A=A0=E6=B5=8B=E8=AF=95):?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 代码回溯 --- app/service/generate_image/service_generate_product_image.py | 2 ++ app/service/generate_image/service_generate_relight_image.py | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/app/service/generate_image/service_generate_product_image.py b/app/service/generate_image/service_generate_product_image.py index fa2340b..06f34d5 100644 --- a/app/service/generate_image/service_generate_product_image.py +++ b/app/service/generate_image/service_generate_product_image.py @@ -243,6 +243,8 @@ class GenerateProductImage: def callback(self, result, error): if error: self.gen_product_data['status'] = "FAILURE" + if 'mask_list' in str(error): + self.gen_product_data['status'] = "NO_FACE" self.gen_product_data['message'] = str(error) self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) else: diff --git a/app/service/generate_image/service_generate_relight_image.py b/app/service/generate_image/service_generate_relight_image.py index e0729ba..3efbe88 100644 --- a/app/service/generate_image/service_generate_relight_image.py +++ b/app/service/generate_image/service_generate_relight_image.py @@ -52,7 +52,11 @@ class GenerateRelightImage: def callback(self, result, error): if error: self.gen_product_data['status'] = "FAILURE" + if 'mask_list' in str(error): + self.gen_product_data['status'] = "NO_FACE" + self.gen_product_data['message'] = str(error) + self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) else: # pil图像转成numpy数组