yolo系列自己画检测框box

import cv2
class Colors:
    # Ultralytics color palette https://ultralytics.com/
    def __init__(self):
        # hex = matplotlib.colors.TABLEAU_COLORS.values()
        hex = ('FF3838', 'FF9D97', 'FF701F', 'FFB21D', 'CFD231', '48F90A', '92CC17', '3DDB86', '1A9334', '00D4BB',
               '2C99A8', '00C2FF', '344593', '6473FF', '0018EC', '8438FF', '520085', 'CB38FF', 'FF95C8', 'FF37C7')
        # 将hex列表中所有hex格式(十六进制)的颜色转换rgb格式的颜色
        self.palette = [self.hex2rgb('#' + c) for c in hex]
        # 颜色个数
        self.n = len(self.palette)

    def __call__(self, i, bgr=False):
        # 根据输入的index 选择对应的rgb颜色
        c = self.palette[int(i) % self.n]
        # 返回选择的颜色 默认是rgb
        return (c[2], c[1], c[0]) if bgr else c

    @staticmethod
    def hex2rgb(h):  # rgb order (PIL)
        # hex -> rgb
        return tuple(int(h[1 + i:1 + i + 2], 16) for i in (0, 2, 4))
def plot_one_box(x, im, color=(128, 128, 128), label=None, line_thickness=2):
    """一般会用在detect.py中在nms之后变量每一个预测框，再将每个预测框画在原图上
    使用opencv在原图im上画一个bounding box
    :params x: 预测得到的bounding box  [x1 y1 x2 y2]
    :params im: 原图 要将bounding box画在这个图上  array
    :params color: bounding box线的颜色
    :params labels: 标签上的框框信息  类别 + score
    :params line_thickness: bounding box的线宽
    """
    # check im内存是否连续
    # tl = 框框的线宽  要么等于line_thickness要么根据原图im长宽信息自适应生成一个
    tl = line_thickness or round(0.002 * (im.shape[0] + im.shape[1]) / 2) + 1  # line/font thickness
    # c1 = (x1, y1) = 矩形框的左上角   c2 = (x2, y2) = 矩形框的右下角
    c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
    # cv2.rectangle: 在im上画出框框   c1: start_point(x1, y1)  c2: end_point(x2, y2)
    # 注意: 这里的c1+c2可以是左上角+右下角  也可以是左下角+右上角都可以
    cv2.rectangle(im, c1, c2, color, thickness=tl, lineType=cv2.LINE_AA)
    # 如果label不为空还要在框框上面显示标签label + score
    # if True:
    tf = max(tl - 1, 1)  # label字体的线宽 font thickness
    # cv2.getTextSize: 根据输入的label信息计算文本字符串的宽度和高度
    # 0: 文字字体类型  fontScale: 字体缩放系数  thickness: 字体笔画线宽
    # 返回retval 字体的宽高 (width, height), baseLine 相对于最底端文本的 y 坐标
    t_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]
    c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3
    # 同上面一样是个画框的步骤  但是线宽thickness=-1表示整个矩形都填充color颜色
    cv2.rectangle(im, c1, c2, color, -1, cv2.LINE_AA)  # filled
    # cv2.putText: 在图片上写文本 这里是在上面这个矩形框里写label + score文本
    # (c1[0], c1[1] - 2)文本左下角坐标  0: 文字样式  fontScale: 字体缩放系数
    # [225, 255, 255]: 文字颜色  thickness: tf字体笔画线宽     lineType: 线样式
    cv2.putText(im, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)
    return im
if __name__ == "__main__":
    img1=cv2.imread(r'')#图片路径
    colors=Colors()
    image1=plot_one_box([280,230,330,390],img1,colors(0),'non:0.75')
    # 添加更多的框和标签
    image1=plot_one_box([335,270,480,370], image1, colors(0), 'non:0.78')  # 新的框和标签
    image1=plot_one_box([253,275,275,300], image1, colors(4), 'int:0.73')  # 新的框和标签
    cv2.imwrite('1.jpg',image1)#保存路径