YOLO数据处理：yolo标签可视化

# -*- coding: utf-8 -*-
import cv2
import numpy as np
from pathlib import Path
import os

#汉字转拼音
from pypinyin import pinyin, Style


def chinese_to_pinyin(text, style=Style.TONE3):
    return ''.join([y[0] for y in pinyin(text, style=style)])
# 类别和颜色信息
id2cls = {
    0: '测试'
}
cls2id = {v: k for k, v in id2cls.items()}
id2color = {i: (np.random.randint(0, 255), np.random.randint(0, 255), np.random.randint(0, 255)) for i in id2cls.keys()}


# 支持中文路径
def cv_imread(filePath):
    cv_img = cv2.imdecode(np.fromfile(filePath, dtype=np.uint8), flags=cv2.IMREAD_COLOR)
    return cv_img


def vis_yolo_single(image_path, label_path, save_path):
    # 读取图像
    image = cv_imread(image_path)
    h, w = image.shape[:2]

    # 检查标签文件是否存在
    if not os.path.exists(label_path):
        print(f"标签文件 {label_path} 不存在！")
        return

    # 读取标签文件
    labels = np.loadtxt(label_path).reshape(-1,