用labelme polygons标出四个点生成的json文件和原图一起放到data文件夹中,同级目录下运行下面的python文件生成txt
import os
import numpy as np
import json
from glob import glob
import cv2
import math
from sklearn.model_selection import train_test_split
from os import getcwd
classes = ["0", "1"]#写自己数据集的标签
labelme_path = "data/" #数据集路径
isUseTest = True # 是否创建test集
files = glob(labelme_path + "*.json")
files = [i.replace("\\", "/").split("/")[-1].split(".json")[0] for i in files]
print(files)
if isUseTest:
trainval_files, test_files = train_test_split(files, test_size=0.1, random_state=55)
else:
trainval_files = files
train_files, val_files = train_test_split(trainval_files, test_size=0.1, random_state=55)
contours=[]
rects=[]
def convert(size, box):
dw = 1. / (size[0])
dh = 1. / (size[1])
x1 = ((box[0] + box[1]) / 2.0 -