OpenCV中如何正确的给文字区域加上底色

OpenCV学堂

发布于 2022-04-28 12:10:11

2.6K0

发布于 2022-04-28 12:10:11

文章被收录于专栏：贾志刚-OpenCV学堂

点击上方蓝字关注我们

微信公众号：OpenCV学堂 关注获取更多计算机视觉与深度学习知识

显示效果

对象检测模型推理解析结果之后，经常需要在对象检测框上部添加文字类别跟其他表述文本，这样显示可以让人一目了然，非常清楚的知道各种检测类别跟自信度信息，但是这个可视化显示，OpenCV可以做的非常好，给人很直观的感觉。图示如下：

如何生成这种显示

OpenCV中有个获取字体跟文本宽高的函数，调用该函数可以获取

Size cv::getTextSize(         const String & text, // 输入文本         int fontFace, // 字体类型         double fontScale, // 字体大小         int thickness, // 字体线宽         int * baseLine // 基线对齐位置)

其中基线对齐位置baseline 参数 是一个很trick的参数，在绘制底色跟绘制文本的时候必须要考虑进去，这个可以看后面的演示代码。

返回的参数类型是cv::Szie文本区域的宽度与长度，有这个就可以根据它完成在文本框上方的文字底色矩形区域绘制，然后在把相关的文本通过putText绘制完成，这样就实现了如下图中显示效果

相关的代码显示如下 - cpp部分：

定义相关变量

int fontface = cv::FONT_HERSHEY_SIMPLEX;int thickness = 1;float fontScale = 0.5;int baseline = 0;Scalar color(140, 199, 0);

使用cv::getTextSize获取文本区域大小并绘制底色矩形，显示检测框:

Rect box((int)tl_x, (int)tl_y, (int)(br_x - tl_x), (int)(br_y - tl_y));rectangle(frame, box, color, 2, 8, 0);cv::String label = format("face, score: %.2f", score);cv::Size fontSize = ::getTextSize(label, fontface, fontScale, thickness, &baseline);cv::rectangle(frame, cv::Size(box.tl().x, box.tl().y - fontSize.height - baseline),         cv::Size(box.tl().x + fontSize.width, box.tl().y), cv::Scalar(34, 43, 59), -1, 8);putText(frame, label, Point(box.tl().x, box.tl().y-baseline), fontface, fontScale, Scalar(255, 0, 255), thickness, 8);

Pytho部分得代码，同样是分为两个部分，实现如下：

# 动态合理显示文本区域 font_color = (140,199,0) cv.rectangle(canvas, (100, 100), (300, 300), font_color, 2, 8) font = cv.FONT_HERSHEY_SIMPLEX font_scale = 0.5 thickness = 1
# 显示部分 (fw, uph), dh = cv.getTextSize("OpenCV-Python", font, font_scale, thickness) cv.rectangle(canvas, (100, 100-uph-dh), (100+fw, 100), (255, 255, 255), -1, 8) cv.putText(canvas, "OpenCV-Python", (100, 100-dh), font, font_scale, (255, 0, 255), thickness) cv.imshow("canvas", canvas) cv.waitKey(0)

扫码查看OpenCV+Pytorch系统化学习路线图