icdar2015标注数据转 yolo-quadrangle数据python脚本

脚本如下

# -*- coding: UTF-8 -*-

# from PIL import Image
import os


def get_filelist(path):
    Filelist = []
    for home, dirs, files in os.walk(path):
        for filename in files:
            # 文件名列表，包含完整路径
            if "txt" in filename:
                Filelist.append(os.path.join(home, filename))
            # # 文件名列表，只包含文件名
            # Filelist.append( filename)

    return Filelist


if __name__ == "__main__":
    filePath = './text_src'
    # 指定保存的文件夹
    outputPath = './text_dst'
    # 获得文件夹下所有文件
    # filePath = './imgs/' # 只能获取当前路径下的文件，不能递归
    # filenames = os.listdir(filePath)
    Filelist = get_filelist(filePath)
    print(len(Filelist))
    # 迭代所有图片
    for filename in Filelist:
        print(filename)
        # 读取图像 标签
        # im = Image.open(filename)
        # 保存
        output_path = filename.replace(filePath, outputPath)
        outputdir = output_path.rsplit('\\', 1)[0]
        if not os.path.exists(outputdir):
            os.mkdir(outputdir)
        file_lineinfo = open(output_path, 'w', encoding='utf-8')

        f = open(filename, encoding='utf-8-sig')
        x_1 = y_1 = x_2 = y_2 = x_3 = y_3 = x_4 = y_4 = 0
        label = 0
        for line in f.readlines():
            print(line)
            data = line.replace('\n', '')
            substr = data.split(',')
            label = "0"
            x_1 = substr[0]
            y_1 = substr[1]
            x_2 = substr[2]
            y_2 = substr[3]
            x_3 = substr[4]
            y_3 = substr[5]
            x_4 = substr[6]
            y_4 = substr[7]
            # content = substr[8]
            line_info = [label, ' ', x_1, ' ', y_1, ' ', x_2, ' ', y_2, ' ', x_3, ' ', y_3, ' ', x_4, ' ', y_4, '\n']
            #line_info = [str(label), x_1, y_1, x_2, y_2, x_3, y_3, x_4, y_4, '\n']
            file_lineinfo.writelines(line_info)
        f.close()
        #line_info = [label, ' ', x_1, ' ', y_1, ' ',  x_2, ' ', y_2, ' ', x_3, ' ', y_3, ' ', x_4, ' ', y_4, '\n']

        file_lineinfo.close()