用yolov8的推理,做了半自动化标签处理。
yolo的标签格式是:
标签 + 每个顶点的坐标
比如:
第一个数字0是类别,后面是每个顶点的坐标,(x,y)这样的。
而且可能是多行的
所以,要把标签画在原图上,看看打的对不对。
画标签到原图上.py
import os
import cv2
import numpy as np
# 图像文件夹路径和标签文件夹路径
image_folder = r'E:\jianceee\images'
label_folder = r'E:\jianceee\labels'
output_folder = r'E:\jianceee\output'
if not os.path.exists(output_folder):
os.makedirs(output_folder)
# 遍历标签文件夹中的每个标签文件
for filename in os.listdir(label_folder):
if filename.endswith('.txt'):
label_path = os.path.join(label_folder, filename)
image_path = os.path.join(image_folder, filename.replace('.txt', '.jpg')) # 假设图像文件与标签文件同名,后缀为.jpg
# 读取原始图像