From d661c3d8c312e7a21ce5447e9d83e82dfa915ca0 Mon Sep 17 00:00:00 2001
From: dyning <dyning.2003@163.com>
Date: Fri, 15 May 2020 22:18:09 +0800
Subject: [PATCH] Update predict_det.py

---
 tools/infer/predict_det.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index e4b3d037..82fed48f 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))
-- 
GitLab