From 42a9a47ce565790b0d561483a22b74245dc988f0 Mon Sep 17 00:00:00 2001 From: tink2123 Date: Tue, 30 Jun 2020 19:28:29 +0800 Subject: [PATCH] close memory_optim for attention --- tools/infer/utility.py | 33 +++++++++++++++++++++------------ 1 file changed, 21 insertions(+), 12 deletions(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 2859c97e..0cc09713 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -91,7 +91,7 @@ def create_predictor(args, mode): else: config.disable_gpu() - config.enable_memory_optim() + #config.enable_memory_optim() config.disable_glog_info() # use zero copy @@ -169,26 +169,35 @@ def draw_ocr_box_txt(image, boxes, txts): 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)) + 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) + 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") + 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) + 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) + 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)) -- GitLab