以下教程用于验证转成YOLO使用的txt格式，适用场景：多边形框，配合json格式文件转成YOLO使用的txt格式脚本使用。
YOLO数据集制作（三）|Labelme标注的“多边形框”json文件转txt-CSDN博客
使用方式：将img_path和label_path分别填入对应的图片（文件夹）及标签（文件夹）路径，运行。show_num参数控制最大展示数量，按空格切换。
import os
import numpy as np
import cv2img_format = ['.jpg', '.png', '.jpeg']
colors = [(0, 0, 0), (128, 0, 0), (0, 128, 0), (128, 128, 0), (0, 0, 128), (128, 0, 128), (0, 128, 128),(128, 128, 128), (64, 0, 0), (192, 0, 0), (64, 128, 0), (192, 128, 0), (64, 0, 128), (192, 0, 128),(64, 128, 128), (192, 128, 128), (0, 64, 0), (128, 64, 0), (0, 192, 0), (128, 192, 0), (0, 64, 128),(128, 64, 12)]def get_files(img_path, label_path):img_list = []label_list = []datast_img_format = Noneassert os.path.exists(img_path) and os.path.exists(label_path), print("⭐⭐文件夹不存在⭐⭐")if os.path.isdir(img_path):for i in os.listdir(img_path):if os.path.splitext(i)[-1] in img_format:# i = i.replace(os.path.splitext(i)[-1], '')img_list.append(i)datast_img_format=img_list[-1].split('.')[-1]if os.path.isdir(label_path):for i in os.listdir(label_path):if os.path.splitext(i)[-1] == '.txt':i = i.replace(os.path.splitext(i)[-1], '')label_list.append(i)print("路径下无jpg,png,jpeg格式的图片,当前图像路径:" + str(img_path) if len(img_list) == 0 else "图像总数为：" + str(len(img_list)))print("路径下无标签文件,当前标签路径" + str(label_path) if len(label_list) == 0 else "标签总数为：" + str(len(label_list)))img_without_label = []for i in img_list:if i.replace(os.path.splitext(i)[-1], '') not in label_list:img_without_label.append(i)if len(img_without_label) != 0:print("标签丢失的图像有：" + str(img_without_label))ok_img = list(set(img_list) - set(img_without_label))return [i.replace('.jpg', '') for i in ok_img], '.'+datast_img_format# 坐标转换
def convert(x, w1, h1, img):label, l = x[0], x[1:]rl = np.array([[float(l[i]) * w1, float(l[i + 1]) * h1] for i in range(0, len(l), 2) if i + 1 < len(l)], dtype=np.int32)# cv2.polylines(img, [rl],color=colors[int(label)] if int(label) < len(colors) else colors[0], thickness=2,isClosed=True)cv2.fillPoly(img, [rl],color=colors[int(label)] if int(label) < len(colors) else colors[0])cv2.putText(img, text=str(int(label)), org=(int(rl[0][0]), int(rl[0][1])), fontFace=cv2.FONT_HERSHEY_COMPLEX,fontScale=1, color=colors[int(label)] if int(label) < len(colors) else colors[0], thickness=3)return imgif __name__ == '__main__':# 修改输入图片文件夹img_path = r"O:\DeepLearningTool\dataset\segment\image\0507.jpg"# img_path = r"O:\DeepLearningTool\dataset\segment\image"# img_path = r"O:\DeepLearningTool\dataset\image"# 修改输入标签文件夹label_path = r"O:\DeepLearningTool\dataset\segment\label\0507.txt"# label_path = r"O:\DeepLearningTool\dataset\segment\label"# label_path = r"O:\DeepLearningTool\dataset\label"# 输出文件夹outfile = './'# 是否展示绘制的图片if_show = True# 最大展示图片的数量(按空格切换)show_num = 3# # 是否保存绘制的图片# if_save = Falseif os.path.isdir(img_path):ok_img, datast_img_format = get_files(img_path, label_path)haveShow = 0for i in ok_img:img = cv2.imread(str(img_path) + '\\' + i + datast_img_format)h, w = img.shape[:2]with open(label_path + '\\' + i + '.txt', 'r') as f:lb = np.array([x.split() for x in f.read().strip().splitlines()], dtype=np.float32)for x in lb:img = convert(x, w, h, img)if if_show:cv2.namedWindow('Image with label', 0)cv2.resizeWindow('Image with label', 600, 500)cv2.imshow('Image with label', img)cv2.waitKey(0)# if if_save:# outfile = outfile if len(outfile) > 0 else './output/'# if not os.path.exists('./output/'):# os.mkdir('./output/')# print(outfile + i)# cv2.imwrite(outfile+i+datast_img_format, img)haveShow += 1if haveShow == show_num:breakelse:img = cv2.imread(str(img_path))h, w = img.shape[:2]assert os.path.isfile(label_path), '标签路径错误'lb = []with open(label_path) as f:lb.append([x.split() for x in f.read().strip().splitlines()])for x in lb[0]:img = convert(x, w, h, img)if if_show:cv2.namedWindow('Image with label', 0)cv2.resizeWindow('Image with label', 600, 500)cv2.imshow('Image with label', img)cv2.waitKey(0))
)
)
