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

feat（新功能）:

Browse Source 
fix（修复bug）:  product generate 获取三通道RGB图片时 无法获取A通道mask 增强图片mode判断逻辑
docs（文档变更）:
refactor（重构）:
test(增加测试):

代码回溯
This commit is contained in:
zhouchengrong
2025-02-12 17:39:36 +08:00
parent 29000d142f
commit 63f0585ce8
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数组