可视化YOLO模型的检测框和关键点的python代码

本篇文章将yolo格式标注文件(txt)中的数据绘制到原图上，实现可视化。文中代码参考了yolov5-face的绘图代码。
https://github.com/deepcam-cn/yolov5-face

从yolov5-face项目中导入需要的文件

from utils.general import xywh2xyxy
from utils.plots import colors,  plot_one_box

完整代码

import torch
import cv2
from utils.general import xywh2xyxy
from utils.plots import colors,  plot_one_box

if __name__ == '__main__':
    colormap = [(255, 255, 255), (56, 56, 255), (151, 157, 255), (31, 112, 255), (29, 178, 255)]   # 不是RGB,是BGR

    # 将yolo格式的标注好的图片展示出来
    im0 = cv2.imread("./face_img/000000191874.jpg")
    f = open(