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-13 13:03:37 +08:00
parent 63f0585ce8
commit 9ba883a6ee
2 changed files with 2 additions and 2 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

@@ -302,7 +302,7 @@ class GenerateProductImage:
time_out = 600 time_out = 600
while time_out > 0: while time_out > 0:
gen_product_data, _ = self.read_tasks_status() gen_product_data, _ = self.read_tasks_status()
if gen_product_data['status'] in ["REVOKED", "FAILURE"]: if gen_product_data['status'] in ["REVOKED", "FAILURE", "NO_FACE"]:
ctx.cancel() ctx.cancel()
break break
elif gen_product_data['status'] == "SUCCESS": elif gen_product_data['status'] == "SUCCESS":

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

@@ -121,7 +121,7 @@ class GenerateRelightImage:
time_out = 600 time_out = 600
while time_out > 0: while time_out > 0:
gen_product_data, _ = self.read_tasks_status() gen_product_data, _ = self.read_tasks_status()
if gen_product_data['status'] in ["REVOKED", "FAILURE"]: if gen_product_data['status'] in ["REVOKED", "FAILURE", "NO_FACE"]:
ctx.cancel() ctx.cancel()
break break
elif gen_product_data['status'] == "SUCCESS": elif gen_product_data['status'] == "SUCCESS":