tuoheng_alg_move_20240715/util/PlotsUtils.py

import cv2
import numpy as np
from PIL import Image, ImageDraw, ImageFont


def get_label_array(color=None, label=None, font=None, fontSize=40):
    x, y, width, height = font.getbbox(label)
    text_image = np.zeros((height, width, 3), dtype=np.uint8)
    text_image = Image.fromarray(text_image)
    draw = ImageDraw.Draw(text_image)
    draw.rectangle((0, 0, width, height), fill=tuple(color))
    draw.text((0, -1), label, fill=(255, 255, 255), font=font)
    im_array = np.asarray(text_image)
    # scale = fontSize / height
    # im_array = cv2.resize(im_array, (0, 0), fx=scale, fy=scale)
    scale = height / fontSize
    im_array = cv2.resize(im_array, (0, 0), fx=scale, fy=scale)
    return im_array


def get_label_arrays(labelNames, colors, fontSize=40, fontPath="platech.ttf"):
    font = ImageFont.truetype(fontPath, fontSize, encoding='utf-8')
    label_arraylist = [get_label_array(colors[i % 20], label_name, font, fontSize) for i, label_name in
                       enumerate(labelNames)]
    return label_arraylist


def xywh2xyxy(box):
    if not isinstance(box[0], (list, tuple, np.ndarray)):
        xc, yc, w, h = int(box[0]), int(box[1]), int(box[2]), int(box[3])
        bw, bh = int(w / 2), int(h / 2)
        lt, yt, rt, yr = xc - bw, yc - bh, xc + bw, yc + bh
        box = [(lt, yt), (rt, yt), (rt, yr), (lt, yr)]
    return box

def xywh2xyxy2(param):
    if not isinstance(param[0], (list, tuple, np.ndarray)):
        xc, yc, x2, y2 = int(param[0]), int(param[1]), int(param[2]), int(param[3])
        return [(xc, yc), (x2, yc), (x2, y2), (xc, y2)], float(param[4]), int(param[5])
        # bw, bh = int(w / 2), int(h / 2)
        # lt, yt, rt, yr = xc - bw, yc - bh, xc + bw, yc + bh
        # return [(lt, yt), (rt, yt), (rt, yr), (lt, yr)]
    return np.asarray(param[0][0:4], np.int32), float(param[1]), int(param[2])


def draw_painting_joint(box, img, label_array, score=0.5, color=None, config=None):
    # 识别问题描述图片的高、宽
    lh, lw = label_array.shape[0:2]
    # 图片的长度和宽度
    imh, imw = img.shape[0:2]
    box = xywh2xyxy(box)
    # 框框左上的位置
    x0, y1 = box[0][0], box[0][1]
    # if score_location == 'leftTop':
    #     x0, y1 = box[0][0], box[0][1]
    # # 框框左下的位置
    # elif score_location == 'leftBottom':
    #     x0, y1 = box[3][0], box[3][1]
    # else:
    #     x0, y1 = box[0][0], box[0][1]
    # x1 框框左上x位置 + 描述的宽
    # y0 框框左上y位置 - 描述的高
    x1, y0 = x0 + lw, y1 - lh
    # 如果y0小于0, 说明超过上边框
    if y0 < 0:
        y0 = 0
        # y1等于文字高度
        y1 = y0 + lh
    # 如果y1框框的高大于图片高度
    if y1 > imh:
        # y1等于图片高度
        y1 = imh
        # y0等于y1减去文字高度
        y0 = y1 - lh
    # 如果x0小于0
    if x0 < 0:
        x0 = 0
        x1 = x0 + lw
    if x1 > imw:
        x1 = imw
        x0 = x1 - lw
    # box_tl = max(int(round(imw / 1920 * 3)), 1) or round(0.002 * (imh + imw) / 2) + 1
    '''
    1. img（array） 为ndarray类型（可以为cv.imread）直接读取的数据
    2. box（array）：为所画多边形的顶点坐标
    3. 所画四边形是否闭合，通常为True
    4. color（tuple）：BGR三个通道的值
    5. thickness（int）：画线的粗细
    6. shift：顶点坐标中小数的位数
    '''
    tl = config[0]
    box1 = np.asarray(box, np.int32)
    cv2.polylines(img, [box1], True, color, tl)
    img[y0:y1, x0:x1, :] = label_array
    pts_cls = [(x0, y0), (x1, y1)]
    # 把英文字符score画到类别旁边
    # tl = max(int(round(imw / 1920 * 3)), 1) or round(0.002 * (imh + imw) / 2) + 1
    label = ' %.2f' % score
    # tf = max(tl, 1)
    # fontScale = float(format(imw / 1920 * 1.1, '.2f')) or tl * 0.33
    # fontScale = tl * 0.33
    '''
    1. text：要计算大小的文本内容，类型为字符串。
    2. fontFace：字体类型，例如cv2.FONT_HERSHEY_SIMPLEX等。
    3. fontScale：字体大小的缩放因子，例如1.2表示字体大小增加20%。
    4. thickness：文本线条的粗细，以像素为单位。
    5. (text_width, text_height)：给定文本在指定字体、字体大小、线条粗细下所占用的像素宽度和高度。
    '''
    # t_size = cv2.getTextSize(label, 0, fontScale=fontScale, thickness=tf)[0]
    t_size = (config[1], config[2])
    # if   socre_location=='leftTop':
    p1, p2 = (pts_cls[1][0], pts_cls[0][1]), (pts_cls[1][0] + t_size[0], pts_cls[1][1])
    '''
    1. img：要绘制矩形的图像
    2. pt1：矩形框的左上角坐标，可以是一个包含两个整数的元组或列表，例如(x1, y1)或[x1, y1]。
    3. pt2：矩形框的右下角坐标，可以是一个包含两个整数的元组或列表，例如(x2, y2)或[x2, y2]。
    4. color：矩形框的颜色，可以是一个包含三个整数的元组或列表，例如(255, 0, 0)表示蓝色，或一个标量值，例如255表示白色。颜色顺序为BGR。
    5. thickness：线条的粗细，以像素为单位。如果为负值，则表示要绘制填充矩形。默认值为1。
    6. lineType：线条的类型，可以是cv2.LINE_AA表示抗锯齿线条，或cv2.LINE_4表示4连通线条，或cv2.LINE_8表示8连通线条。默认值为cv2.LINE_8。
    7. shift：坐标点小数点位数。默认值为0。
    '''
    cv2.rectangle(img, p1, p2, color, -1, cv2.LINE_AA)
    p3 = pts_cls[1][0], pts_cls[1][1] - (lh - t_size[1]) // 2
    '''
    1. img：要在其上绘制文本的图像
    2. text：要绘制的文本内容，类型为字符串
    3. org：文本起始位置的坐标，可以是一个包含两个整数的元组或列表，例如(x, y)或[x, y]。
    4. fontFace：字体类型，例如cv2.FONT_HERSHEY_SIMPLEX等。
    5. fontScale：字体大小的缩放因子，例如1.2表示字体大小增加20%。
    6. color：文本的颜色，可以是一个包含三个整数的元组或列表，例如(255, 0, 0)表示蓝色，或一个标量值，例如255表示白色。颜色顺序为BGR。
    7. thickness：文本线条的粗细，以像素为单位。默认值为1。
    8. lineType：线条的类型，可以是cv2.LINE_AA表示抗锯齿线条，或cv2.LINE_4表示4连通线条，或cv2.LINE_8表示8连通线条。默认值为cv2.LINE_8。
    9. bottomLeftOrigin：文本起始位置是否为左下角。如果为True，则文本起始位置为左下角，否则为左上角。默认值为False。
    '''
    cv2.putText(img, label, p3, 0, config[3], [225, 255, 255], thickness=config[4], lineType=cv2.LINE_AA)
    return img, box