forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

make predict_det.py work on windows (#1236) 

* make predict_det.py work on windows
This commit is contained in:
Mike Shi 2020-11-26 21:33:44 +08:00 committed by GitHub
parent 2ffbece713
commit ca5308c94e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/infer/predict_det.py
View File

@ -177,7 +177,7 @@ if __name__ == "__main__":
text_detector = TextDetector(args) text_detector = TextDetector(args)
count = 0 count = 0
total_time = 0 total_time = 0
draw_img_save = "./inference_results" draw_img_save = os.path.join(".", "inference_results")
if not os.path.exists(draw_img_save): if not os.path.exists(draw_img_save):
os.makedirs(draw_img_save) os.makedirs(draw_img_save)
for image_file in image_file_list: for image_file in image_file_list:
@ -193,8 +193,8 @@ if __name__ == "__main__":
count += 1 count += 1
print("Predict time of %s:" % image_file, elapse) print("Predict time of %s:" % image_file, elapse)
src_im = utility.draw_text_det_res(dt_boxes, image_file) src_im = utility.draw_text_det_res(dt_boxes, image_file)
img_name_pure = image_file.split("/")[-1] img_name_pure = os.path.split(image_file)[-1] # image_file.split("/")[-1]
cv2.imwrite( img_path = os.path.join(draw_img_save, "det_res_%s" % img_name_pure)
os.path.join(draw_img_save, "det_res_%s" % img_name_pure), src_im) cv2.imwrite(img_path, src_im)
if count > 1: if count > 1:
print("Avg Time:", total_time / (count - 1)) print("Avg Time:", total_time / (count - 1))