Skip to content

GitLab

Commit a44c2a69 authored by WenmuZhou's avatar WenmuZhou
Browse files

Adapt to windows

parent bbc52fc9
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tools/infer/predict_det.py
View file @ a44c2a69
...@@ -180,8 +180,8 @@ if __name__ == "__main__": ...@@ -180,8 +180,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]
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))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment