diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index e4b3d037779376aeb3100b69bd103a64b75d73c2..82fed48feec063fc4379bc469372f876aabaad6f 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -139,6 +139,8 @@ if __name__ == "__main__": total_time += elapse count += 1 print("Predict time of %s:" % image_file, elapse) - 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] + cv2.imwrite("./inference_results/det_res_%s" % img_name_pure, src_im) if count > 1: print("Avg Time:", total_time / (count - 1))