From 6dd8da089f679888b8fda73acbd4390023473dc3 Mon Sep 17 00:00:00 2001 From: andyjpaddle Date: Thu, 29 Sep 2022 07:51:13 +0000 Subject: [PATCH] format code --- tools/infer/predict_det.py | 8 +++----- tools/infer/utility.py | 10 +++------- 2 files changed, 6 insertions(+), 12 deletions(-) diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 7e371886..52c225d2 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 5de672be..e555dbec 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): -- GitLab