背景简述
我们有一批标注项目要转可视化,因为之前没有做过,然后网上随意找了一段代码测试完美(并没有)搞定,开始疯狂标注,当真正要转的时候傻眼了,因为测试的时候用的是英文标签,实际标注的是中文标签,结果都是一大堆??????,
结果瞬间让我满脑袋??????,赶紧找资料解决,各种方法试了个遍,网上大多数都是用cv2+matplotlib实现的计算和渲染,所以解决的主要思想就是集中在各种显示的设置matplotlib字体,然并卵;最后找到一种另辟蹊径的办法使用PIL+cv2实现,最后完美解决,
贴上解决代码:
import cv2
import os
import numpy as np
from PIL import Image, ImageDraw, ImageFont
import xml.etree.ElementTree as ET
data_path = 'E:\\test\\tianjingulou'
imgs_path = os.path.join(data_path, "img")
anns_path = os.path.join(data_path, "xml")
result_path = os.path.join(data_path)
img_names = set([os.path.splitext(i)[0] for i in os.listdir(imgs_path)])
ann_names = set([os.path.splitext(i)[0] for i in os.listdir(anns_path)])
img_names = list(img_names)
ann_names = list(ann_names)
for i in range(len(img_names)):
img_path = os.path.join(imgs_path, img_names[i] + ".jpg")
img_bgr = cv2.imread(img_path)
xml_path = os.path.join(anns_path, ann_names[i] + ".xml")
xml_inf = open(xml_path, encoding='utf-8')
tree = ET.parse(xml_inf)
root = tree.getroot()
bbox_color = (0, 129, 255)
bbox_thickness = 2
# 把rgb转成16进制'#0081FF'
bbox_color_str = "#{:02x}{:02x}{:02x}".format(*bbox_color)
# 把rgb转成bgr再转16进制'#FF8100'
# bbox_color_rgb = bbox_color[::-1]
# bbox_color_str = "#{:02x}{:02x}{:02x}".format(*bbox_color_rgb)
bbox_labelstr = {
'font_size': 16,
'font_thickness': 2,
'offset_x': 0,
'offset_y': -20,
}