diff --git a/paddlehub/serving/app_single.py b/paddlehub/serving/app_single.py index 6ef2378af2a58983a6c1f27f40e1adf088ffdb2e..c19a4ff89fcc231e1cfb920bc4df98a2a7c1110a 100644 --- a/paddlehub/serving/app_single.py +++ b/paddlehub/serving/app_single.py @@ -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" @@ -232,6 +233,7 @@ def predict_semantic_segmentation(module, input_img, id, batch_size): b_body = str(b_body).replace("b'", "").replace("'", "") b_img = b_head + "," + b_body base64_list.append(b_img) + output_file_path = results[index]["processed"] results[index]["origin"] = results[index]["origin"].replace( id + "_", "") results[index]["processed"] = results[index]["processed"].replace( @@ -239,7 +241,8 @@ def predict_semantic_segmentation(module, input_img, id, batch_size): 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