diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index efda0e46cf37aec97b954fb829e8053573631825..c395502bb020b182577c48710130d1e6f5aa0982 100755
--- a/tools/infer/predict_det.py
+++ b/tools/infer/predict_det.py
@@ -180,8 +180,8 @@ if __name__ == "__main__":
         count += 1
         print("Predict time of %s:" % image_file, elapse)
         src_im = utility.draw_text_det_res(dt_boxes, image_file)
-        img_name_pure = image_file.split("/")[-1]
-        cv2.imwrite(
-            os.path.join(draw_img_save, "det_res_%s" % img_name_pure), src_im)
+        img_name_pure = os.path.split(image_file)[-1]
+        img_path = os.path.join(draw_img_save, "det_res_%s" % img_name_pure)
+        cv2.imwrite(img_path, src_im)
     if count > 1:
         print("Avg Time:", total_time / (count - 1))