提取labelme标注文件信息(json转txt)文本检测四边形坐标标注提取

功能说明

将labelme标注的json文件中的坐标和label信息提取到txt文件中,一般用于检测类任务
ps:这里使用“”四边形+类别”标注方式,每个点坐标包括x和y,所以总共输出8个坐标值和1个label值

软件

labelme 3.16.7

目录结构

# OS找得到就行
dir_json
dir_txt
json2txt.py

json格式

提取labelme标注文件信息(json转txt)文本检测四边形坐标标注提取_第1张图片
“shapes"中包含多个"label”
末尾有图片信息等
提取labelme标注文件信息(json转txt)文本检测四边形坐标标注提取_第2张图片

单个json文件提取坐标至txt

def json2txt(path_json,path_txt):
    with open(path_json,'r') as path_json:
        jsonx=json.load(path_json)
        with open(path_txt,'w+') as ftxt:
            for shape in jsonx['shapes']:           
                xy=np.array(shape['points'])
                label=str(shape['label'])
                strxy = ''
                for m,n in xy:
                    strxy+=str(m)+','+str(n)+','
                strxy+=label                                            
                ftxt.writelines(strxy+"\n")  

批量转换

dir_json为json文件夹
dir_txt为txt文件夹

dir_json = 'jsonlabel/'
dir_txt = 'txtlabel/'
if not os.path.exists(dir_txt):
    os.makedirs(dir_txt)
list_json = os.listdir(dir_json)
for cnt,json_name in enumerate(list_json):
    print('cnt=%d,name=%s'%(cnt,json_name))
    path_json = dir_json + json_name
    path_txt = dir_txt + json_name.replace('.json','.txt')
    #print(path_json,path_txt)    
    json2txt(path_json,path_txt)

输出示例

dir_json中有类似“1.json”的标注文件
运行后在dir_txt中生成“1.txt”格式的标注文件
每一行表示{x1,y1,x2,y2,x3,y3,x4,y4,label}
x,y即之前json中points的坐标点
提取labelme标注文件信息(json转txt)文本检测四边形坐标标注提取_第3张图片

你可能感兴趣的:(深度学习,json,神经网络)