mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

format code

pull/7775/head
andyjpaddle 2022-09-29 07:51:13 +00:00
parent e348e7fe9d
commit 6dd8da089f
2 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/infer/predict_det.py
View File

@ -326,11 +326,9 @@ if __name__ == "__main__":
else: else:
logger.info("{} The predict time of {}: {}".format( logger.info("{} The predict time of {}: {}".format(
idx, image_file, elapse)) idx, image_file, elapse))
if flag_pdf:
src_im = utility.draw_text_det_res(dt_boxes, img, flag_pdf) src_im = utility.draw_text_det_res(dt_boxes, img)
else:
src_im = utility.draw_text_det_res(dt_boxes, image_file,
flag_pdf)
if flag_gif: if flag_gif:
save_file = image_file[:-3] + "png" save_file = image_file[:-3] + "png"
elif flag_pdf: elif flag_pdf:

10
tools/infer/utility.py
View File

@ -338,15 +338,11 @@ def draw_e2e_res(dt_boxes, strs, img_path):
return src_im return src_im
def draw_text_det_res(dt_boxes, img_path, flag_pdf=False): def draw_text_det_res(dt_boxes, img):
if not flag_pdf:
src_im = cv2.imread(img_path)
else:
src_im = img_path
for box in dt_boxes: for box in dt_boxes:
box = np.array(box).astype(np.int32).reshape(-1, 2) box = np.array(box).astype(np.int32).reshape(-1, 2)
cv2.polylines(src_im, [box], True, color=(255, 255, 0), thickness=2) cv2.polylines(img, [box], True, color=(255, 255, 0), thickness=2)
return src_im return img
def resize_img(img, input_size=600): def resize_img(img, input_size=600):