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

Adapt to windows

This commit is contained in:
WenmuZhou 2020-11-30 16:48:51 +08:00
parent bbc52fc993
commit a44c2a699f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/infer/predict_det.py
View File

@ -180,8 +180,8 @@ if __name__ == "__main__":
count += 1
print("Predict time of %s:" % image_file, elapse)
src_im = utility.draw_text_det_res(dt_boxes, image_file)
img_name_pure = image_file.split("/")[-1]
cv2.imwrite(
os.path.join(draw_img_save, "det_res_%s" % img_name_pure), src_im)
img_name_pure = os.path.split(image_file)[-1]
img_path = os.path.join(draw_img_save, "det_res_%s" % img_name_pure)
cv2.imwrite(img_path, src_im)
if count > 1:
print("Avg Time:", total_time / (count - 1))