diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index 7e371886f1478871fe2620ede4eac6da594689a2..52c225d2b3913cf8c0dc88abcc07f7ccfd3cc914 100755
--- a/tools/infer/predict_det.py
+++ b/tools/infer/predict_det.py
@@ -326,11 +326,9 @@ if __name__ == "__main__":
             else:
                 logger.info("{} The predict time of {}: {}".format(
                     idx, image_file, elapse))
-            if flag_pdf:
-                src_im = utility.draw_text_det_res(dt_boxes, img, flag_pdf)
-            else:
-                src_im = utility.draw_text_det_res(dt_boxes, image_file,
-                                                   flag_pdf)
+
+            src_im = utility.draw_text_det_res(dt_boxes, img)
+
             if flag_gif:
                 save_file = image_file[:-3] + "png"
             elif flag_pdf:
diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index 5de672be4c8e5c69dd2c67c2dc011541b9fe40ea..e555dbec1b314510aaaf6b31f1b35bf60fefa98e 100644
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -338,15 +338,11 @@ def draw_e2e_res(dt_boxes, strs, img_path):
     return src_im
 
 
-def draw_text_det_res(dt_boxes, img_path, flag_pdf=False):
-    if not flag_pdf:
-        src_im = cv2.imread(img_path)
-    else:
-        src_im = img_path
+def draw_text_det_res(dt_boxes, img):
     for box in dt_boxes:
         box = np.array(box).astype(np.int32).reshape(-1, 2)
-        cv2.polylines(src_im, [box], True, color=(255, 255, 0), thickness=2)
-    return src_im
+        cv2.polylines(img, [box], True, color=(255, 255, 0), thickness=2)
+    return img
 
 
 def resize_img(img, input_size=600):