修改显示方式,微调显示参数
This commit is contained in:
parent
3f402caea7
commit
22e8cf8c6b
|
@ -134,9 +134,8 @@ if __name__ == "__main__":
|
||||||
txts = [rec_res[i][0] for i in range(len(rec_res))]
|
txts = [rec_res[i][0] for i in range(len(rec_res))]
|
||||||
scores = [rec_res[i][1] for i in range(len(rec_res))]
|
scores = [rec_res[i][1] for i in range(len(rec_res))]
|
||||||
|
|
||||||
# draw_img = draw_ocr(
|
draw_img = draw_ocr(
|
||||||
# image, boxes, txts, scores, draw_txt=True, drop_score=0.5)
|
image, boxes, txts, scores, draw_txt=True, drop_score=0.5)
|
||||||
draw_img = draw_ocr_box_txt(image, boxes, txts)
|
|
||||||
draw_img_save = "./inference_results/"
|
draw_img_save = "./inference_results/"
|
||||||
if not os.path.exists(draw_img_save):
|
if not os.path.exists(draw_img_save):
|
||||||
os.makedirs(draw_img_save)
|
os.makedirs(draw_img_save)
|
||||||
|
|
|
@ -157,20 +157,23 @@ def draw_ocr(image, boxes, txts, scores, draw_txt=True, drop_score=0.5):
|
||||||
return image
|
return image
|
||||||
|
|
||||||
|
|
||||||
def draw_ocr_box_txt(image, boxes, txts, drop_score=0.5):
|
def draw_ocr_box_txt(image, boxes, txts):
|
||||||
img_show = Image.new('RGB', (image.width * 2, image.height), (255, 255, 255))
|
h, w = image.height, image.width
|
||||||
img_show.paste(image, (image.width, 0, image.width * 2, image.height))
|
img_left = image.copy()
|
||||||
img_show_bak = img_show.copy()
|
img_right = Image.new('RGB', (w, h), (255, 255, 255))
|
||||||
height, width = image.height, image.width
|
|
||||||
|
|
||||||
import random
|
import random
|
||||||
random.seed(12345)
|
# 每次使用相同的随机种子 ,可以保证两次颜色一致
|
||||||
draw = ImageDraw.Draw(img_show)
|
random.seed(0)
|
||||||
|
draw_left = ImageDraw.Draw(img_left)
|
||||||
|
draw_right = ImageDraw.Draw(img_right)
|
||||||
for (box, txt) in zip(boxes, txts):
|
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.polygon(box, outline=color)
|
draw_left.polygon(box, fill=color)
|
||||||
draw.polygon([width + box[0][0], box[0][1], width + box[1][0], box[1][1], width + box[2][0], box[2][1], width + box[3][0], box[3][1]],
|
draw_right.polygon([box[0][0], box[0][1],
|
||||||
fill=color)
|
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_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)
|
box_width = math.sqrt((box[0][0] - box[1][0]) ** 2 + (box[0][1] - box[1][1]) ** 2)
|
||||||
if box_height > 2 * box_width:
|
if box_height > 2 * box_width:
|
||||||
|
@ -179,13 +182,16 @@ def draw_ocr_box_txt(image, boxes, txts, drop_score=0.5):
|
||||||
cur_y = box[0][1]
|
cur_y = box[0][1]
|
||||||
for c in txt:
|
for c in txt:
|
||||||
char_size = font.getsize(c)
|
char_size = font.getsize(c)
|
||||||
draw.text((box[0][0] + 3, cur_y), c, fill=(0, 0, 255), font=font)
|
draw_right.text((box[0][0] + 3, cur_y), c, fill=(0, 0, 0), font=font)
|
||||||
cur_y += char_size[1]
|
cur_y += char_size[1]
|
||||||
else:
|
else:
|
||||||
font_size = max(int(box_height * 0.8), 10)
|
font_size = max(int(box_height * 0.8), 10)
|
||||||
font = ImageFont.truetype("./doc/simfang.ttf", font_size, encoding="utf-8")
|
font = ImageFont.truetype("./doc/simfang.ttf", font_size, encoding="utf-8")
|
||||||
draw.text(box[0], txt, fill=(255, 0, 0), font=font)
|
draw_right.text([box[0][0], box[0][1]], txt, fill=(0, 0, 0), font=font)
|
||||||
img_show = Image.blend(img_show_bak, img_show, 0.75)
|
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)
|
return np.array(img_show)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue