python OpenCV画 bounding box并标明数据类

咸鱼丶的自我修养

于 2020-04-26 20:36:22 发布

阅读量4.6k

点赞数 5

分类专栏： python 文章标签： opencv python 计算机视觉

本文链接：https://blog.csdn.net/qq_37333800/article/details/105776456

版权

python 专栏收录该内容

5 篇文章

订阅专栏

本文展示如何使用Python和OpenCV库在图像上绘制边界框(b_box)并添加类别名称，涉及图像读取、窗口创建、矩形绘制、文本放置等关键步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import cv2

    class_name = "car"
    # b_box 左上角坐标
    ptLeftTop = np.array([40, 145])
    # 文本框左上角坐标
    textleftop = []
    # b_box 右下角坐标
    ptRightBottom =np.array([365, 400])
    # 框的颜色
    point_color = (0, 255, 0)
    # 线的厚度
    thickness = 2
    # 线的类型
    lineType = 4

    # (500, 375, 3) -> h w c
    src = cv2.imread('data/detect/000020.jpg')
    cv2.namedWindow('image', cv2.WINDOW_AUTOSIZE)
    src = np.array(src)
    # 画 b_box
    cv2.rectangle(src, tuple(ptLeftTop), tuple(ptRightBottom), point_color, thickness, lineType)

    # 获取文字区域框大小
    t_size = cv2.getTextSize(class_name, 1, cv2.FONT_HERSHEY_PLAIN, 1)[0]
    # 获取 文字区域右下角坐标
    textlbottom = ptLeftTop + np.array(list(t_size))
    # 绘制文字区域矩形框
    cv2.rectangle(src, tuple(ptLeftTop), tuple(textlbottom),  point_color, -1)
    # 计算文字起始位置偏移
    ptLeftTop[1] = ptLeftTop[1] + (t_size[1]/2 + 4)
    # 绘字
    cv2.putText(src, class_name , tuple(ptLeftTop), cv2.FONT_HERSHEY_PLAIN, 1.0, (255, 0, 255), 1)
    # 打印图片的shape
    print(src.shape)

    cv2.imshow('image', src)
    cv2.waitKey(0)
    cv2.destroyAllWindows()