提交 0ab13cbf 编写于 作者: S shenyuhan

1.add config batch_size; 2.delete req_id_ for every file

上级 08e7be50
......@@ -225,6 +225,7 @@ def predict_semantic_segmentation(module, input_img, id, batch_size):
for index in range(len(input_img)):
# special
item = input_img[index]
output_file_path = ""
with open(results[index]["processed"], "rb") as fp:
# special
b_head = "data:image/png;base64"
......@@ -234,12 +235,14 @@ def predict_semantic_segmentation(module, input_img, id, batch_size):
base64_list.append(b_img)
results[index]["origin"] = results[index]["origin"].replace(
id + "_", "")
output_file_path = results[index]["processed"]
results[index]["processed"] = results[index]["processed"].replace(
id + "_", "")
results[index].update({"base64": b_img})
results_pack.append(results[index])
os.remove(item)
os.remove(results[index]["processed"])
if output_file_path != "":
os.remove(output_file_path)
return results_pack
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册