diff --git a/doc/imgs_results_vis2/1.jpg b/doc/imgs_results_vis2/1.jpg new file mode 100644 index 00000000..f6bb48b6 Binary files /dev/null and b/doc/imgs_results_vis2/1.jpg differ diff --git a/doc/imgs_results_vis2/10.jpg b/doc/imgs_results_vis2/10.jpg new file mode 100644 index 00000000..03585245 Binary files /dev/null and b/doc/imgs_results_vis2/10.jpg differ diff --git a/doc/imgs_results_vis2/11.jpg b/doc/imgs_results_vis2/11.jpg new file mode 100644 index 00000000..2b5392a5 Binary files /dev/null and b/doc/imgs_results_vis2/11.jpg differ diff --git a/doc/imgs_results_vis2/12.jpg b/doc/imgs_results_vis2/12.jpg new file mode 100644 index 00000000..a7b6518c Binary files /dev/null and b/doc/imgs_results_vis2/12.jpg differ diff --git a/doc/imgs_results_vis2/13.png b/doc/imgs_results_vis2/13.png new file mode 100644 index 00000000..fca7ac39 Binary files /dev/null and b/doc/imgs_results_vis2/13.png differ diff --git a/doc/imgs_results_vis2/15.jpg b/doc/imgs_results_vis2/15.jpg new file mode 100644 index 00000000..47a32eef Binary files /dev/null and b/doc/imgs_results_vis2/15.jpg differ diff --git a/doc/imgs_results_vis2/16.png b/doc/imgs_results_vis2/16.png new file mode 100644 index 00000000..191c4759 Binary files /dev/null and b/doc/imgs_results_vis2/16.png differ diff --git a/doc/imgs_results_vis2/17.png b/doc/imgs_results_vis2/17.png new file mode 100644 index 00000000..0ba1b073 Binary files /dev/null and b/doc/imgs_results_vis2/17.png differ diff --git a/doc/imgs_results_vis2/2.jpg b/doc/imgs_results_vis2/2.jpg new file mode 100644 index 00000000..8e46314c Binary files /dev/null and b/doc/imgs_results_vis2/2.jpg differ diff --git a/doc/imgs_results_vis2/22.jpg b/doc/imgs_results_vis2/22.jpg new file mode 100644 index 00000000..aea7791c Binary files /dev/null and b/doc/imgs_results_vis2/22.jpg differ diff --git a/doc/imgs_results_vis2/3.jpg b/doc/imgs_results_vis2/3.jpg new file mode 100644 index 00000000..f3ac72ac Binary files /dev/null and b/doc/imgs_results_vis2/3.jpg differ diff --git a/doc/imgs_results_vis2/4.jpg b/doc/imgs_results_vis2/4.jpg new file mode 100644 index 00000000..6986959d Binary files /dev/null and b/doc/imgs_results_vis2/4.jpg differ diff --git a/doc/imgs_results_vis2/5.jpg b/doc/imgs_results_vis2/5.jpg new file mode 100644 index 00000000..c6fadd0e Binary files /dev/null and b/doc/imgs_results_vis2/5.jpg differ diff --git a/doc/imgs_results_vis2/6.jpg b/doc/imgs_results_vis2/6.jpg new file mode 100644 index 00000000..232e53a3 Binary files /dev/null and b/doc/imgs_results_vis2/6.jpg differ diff --git a/doc/imgs_results_vis2/7.jpg b/doc/imgs_results_vis2/7.jpg new file mode 100644 index 00000000..31ee1078 Binary files /dev/null and b/doc/imgs_results_vis2/7.jpg differ diff --git a/doc/imgs_results_vis2/8.jpg b/doc/imgs_results_vis2/8.jpg new file mode 100644 index 00000000..db6e913c Binary files /dev/null and b/doc/imgs_results_vis2/8.jpg differ diff --git a/doc/imgs_results_vis2/9.jpg b/doc/imgs_results_vis2/9.jpg new file mode 100644 index 00000000..5b28f97b Binary files /dev/null and b/doc/imgs_results_vis2/9.jpg differ diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 032d094c..57f9a7e0 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -30,6 +30,7 @@ import time from ppocr.utils.utility import get_image_file_list from PIL import Image from tools.infer.utility import draw_ocr +from tools.infer.utility import draw_ocr_box_txt class TextSystem(object): diff --git a/tools/infer/utility.py b/tools/infer/utility.py index c3d3b445..ba5a3f38 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -157,6 +157,44 @@ def draw_ocr(image, boxes, txts, scores, draw_txt=True, drop_score=0.5): return image +def draw_ocr_box_txt(image, boxes, txts): + h, w = image.height, image.width + img_left = image.copy() + img_right = Image.new('RGB', (w, h), (255, 255, 255)) + + import random + # 每次使用相同的随机种子 ，可以保证两次颜色一致 + random.seed(0) + draw_left = ImageDraw.Draw(img_left) + draw_right = ImageDraw.Draw(img_right) + for (box, txt) in zip(boxes, txts): + color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)) + draw_left.polygon(box, fill=color) + draw_right.polygon([box[0][0], box[0][1], + box[1][0], box[1][1], + box[2][0], box[2][1], + box[3][0], box[3][1]], outline=color) + box_height = math.sqrt((box[0][0] - box[3][0]) ** 2 + (box[0][1] - box[3][1]) ** 2) + box_width = math.sqrt((box[0][0] - box[1][0]) ** 2 + (box[0][1] - box[1][1]) ** 2) + if box_height > 2 * box_width: + font_size = max(int(box_width * 0.9), 10) + font = ImageFont.truetype("./doc/simfang.ttf", font_size, encoding="utf-8") + cur_y = box[0][1] + for c in txt: + char_size = font.getsize(c) + draw_right.text((box[0][0] + 3, cur_y), c, fill=(0, 0, 0), font=font) + cur_y += char_size[1] + else: + font_size = max(int(box_height * 0.8), 10) + font = ImageFont.truetype("./doc/simfang.ttf", font_size, encoding="utf-8") + draw_right.text([box[0][0], box[0][1]], txt, fill=(0, 0, 0), font=font) + img_left = Image.blend(image, img_left, 0.5) + img_show = Image.new('RGB', (w * 2, h), (255, 255, 255)) + img_show.paste(img_left, (0, 0, w, h)) + img_show.paste(img_right, (w, 0, w * 2, h)) + return np.array(img_show) + + def str_count(s): """ Count the number of Chinese characters,