YOLO语义分割标注文件txt还原到图像中

最近做图像分割任务过程中，使用labelme对图像进行标注，得到的数据文件是json，转换为YOLO训练所需的txt格式后，想对标注文件进行检验，即将txt标注文件还原到原图像中，下面是代码：

import cv2
import numpy as np
'''
读取txt标注文件及原图
'''
def read_txt_labels(txt_file):
  """
  从 txt 标注文件中读取标签
  :param txt_file: txt 标注文件路径
  :return: 标签列表
  """
  with open(txt_file, "r") as f:
    labels = []
    for line in f.readlines():
      label_data = line.strip().split(" ")
      class_id = int(label_data[0])
      # 解析边界框坐标
      coordinates = [float(x) for x in label_data[1:]]
      labels.append([class_id, coordinates])
  return labels

def draw_labels</