Update predict_det.py

2020-05-15 22:18:09 +08:00 · 2020-05-15 22:18:09 +08:00 · d661c3d8c3
parent c7ef62df04
commit d661c3d8c3
1 changed files with 3 additions and 1 deletions
--- 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))