提交 8419f9d5 编写于 作者: 走神的阿圆's avatar 走神的阿圆 提交者: wuzewu

fix file not exist (#214)

* add __init__ in templates

* modify setup.py

* delete filename in app, modify main.html to get modules from back

* add single_app()

* add single_app()

* add single_app()

* add single_app()

* add single_app()

* 1.add 'print predict error',2.modify simnet_bow handle func

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

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

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