feat(新功能):

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

View File

@@ -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数组