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

Merge branch 'refs/heads/develop'

Browse Source
This commit is contained in:
zhouchengrong
2025-02-12 17:44:50 +08:00
parent d7832367b4 63f0585ce8
commit d26f38359c
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -243,6 +243,8 @@ class GenerateProductImage:
def callback(self, result, error): def callback(self, result, error):
if error: if error:
self.gen_product_data['status'] = "FAILURE" 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.gen_product_data['message'] = str(error)
self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data))
else: else:

4
app/service/generate_image/service_generate_relight_image.py
View File

@@ -52,7 +52,11 @@ class GenerateRelightImage:
def callback(self, result, error): def callback(self, result, error):
if error: if error:
self.gen_product_data['status'] = "FAILURE" 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.gen_product_data['message'] = str(error)
self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data))
else: else:
# pil图像转成numpy数组 # pil图像转成numpy数组