LabelMe 标注的json转txt的格式转换教程

1 labelme标注后会生成json文件

2 json解析

• annotation里面存放labelme生成的标注文件
• imgs里面存图片
• txtlabel文件夹存放我们转换后的txt标签文件、
• 

  直接建一个和上面一样的文件夹

3  从json查看标注结果

import json
import os
import cv2
img_folder_path=r'F:\imagedata\FOV\FOV_SRC_2\imgs'#原图片存放地址
folder_path=r"F:\imagedata\FOV\FOV_SRC_2\annotation"#标注数据的文件地址


# cv2.putText(img, str, (123,456)), font, 2, (0,255,0), 3)
# 各参数依次是：图片，添加的文字，左上角坐标，字体，字体大小，颜色，字体粗细
def show_label_from_json(img_path,json_d):
    window_name = ('src') 
    cv2.namedWindow(window_name,cv2.WINDOW_FREERATIO)
    src_img=cv2.imread(img_path)   
    font = cv2.FONT_HERSHEY_SIMPLEX
    for item in json_d["shapes"]:
        #print(item['points'])
        point=item['points']
        p1=(int(point[0][0]),int(point[0][1]))
        p2=(int(point[1][0]),int(point[1][1]))
        cv2.rectangle(src_img, p1, p2, (0, 255, 0), 2)
        cv2.putText(src_img,item['label'],p1,font,2,(0,255,255),3)
        cv2.imshow(window_name,src_img)
        cv2.waitKey(0)
    cv2.destroyAllWindows()
    return 

i=0
for jsonfile in os.listdir(folder_path):
    temp_path=os.path.join(folder_path,jsonfile)
    
    i+=1
    if i>5:
        break
    #如果是一个子目录就继续
    if os.path.isdir(temp_path):
        continue
    print("json_path:\t",temp_path)
    temp_path
    with open(temp_path, "r", encoding='utf-8') as f:
        json_d = json.load(f)
        img_name=json_d['imagePath'].split("\\")[-1].split(".")[0]+".jpg"
        img_path=os.path.join(img_folder_path,img_name)    
        print("img_path:\t",img_path)
        show_label_from_json(img_path,json_d)

4 json转txt

有些用图片的绝对坐标，有的用相对坐标，下面两种方式都写了

• 相对坐标形式 :label x_center y_center w h
• 绝对坐标形式 :label x1 y1 x2 y2（x1,y1是标记的左上角的坐标，x2 y2右下角坐标）

import json
import os
import cv2
img_folder_path=r'F:\imagedata\FOV\FOV_SRC_2\imgs'#图片存放文件夹
folder_path=r"F:\imagedata\FOV\FOV_SRC_2\annotation"#标注数据的文件地址
txt_folder_path = r"F:\imagedata\FOV\FOV_SRC_2\txtlabel"#转换后的txt标签文件存放的文件夹

#保存为相对坐标形式 :label x_center y_center w h
def relative_coordinate_txt(img_name,json_d,img_path):
    src_img=cv2.imread(img_path)
    h,w = src_img.shape[:2]
    txt_name = img_name.split(".")[0]+".txt"
    txt_path = os.path.join(txt_folder_path,txt_name)
    print(txt_path)
    with open(txt_path,'w') as f:
        for item in json_d["shapes"]:
            #print(item['points'])
            #print(item['label'])
            point=item['points']
            x_center = (point[0][0]+point[1][0])/2
            y_center = (point[0][1]+point[1][1])/2
            width = point[1][0]-point[0][0]
            hight = point[1][1]-point[0][1]
            #print(x_center)
            f.write(" {} ".format(item['label']))
            f.write(" {} ".format(x_center/w))
            f.write(" {} ".format(y_center/h))
            f.write(" {} ".format(width/w))
            f.write(" {} ".format(hight/h))
            f.write(" \n")
            
#保存为绝对坐标形式 :label x1 y1 x2 y2
def absolute_coordinate_txt(img_name,json_d,img_path):
    src_img=cv2.imread(img_path)
    h,w = src_img.shape[:2]
    txt_name = img_name.split(".")[0]+".txt"
    txt_path = os.path.join(txt_folder_path,txt_name)
    print("txt_path:\t",txt_path)
    with open(txt_path,'w') as f:
        for item in json_d["shapes"]:
            #print(item['points'])
            #print(item['label'])
            point=item['points']
            x1 = point[0][0]
            y1 = point[0][1]
            x2 = point[1][0]
            y2 = point[1][1]
            f.write(" {} ".format(item['label']))
            f.write(" {} ".format(x1))
            f.write(" {} ".format(y1))
            f.write(" {} ".format(x2))
            f.write(" {} ".format(y2))
            f.write(" \n")

i=0
for jsonfile in os.listdir(folder_path):
    temp_path=os.path.join(folder_path,jsonfile)
    
    i+=1
    if i>5:
        break
    #如果是一个子目录就继续
    if os.path.isdir(temp_path):
        continue
    print("json_path:\t",temp_path)
    jsonfile_path=temp_path
    with open(jsonfile_path, "r", encoding='utf-8') as f:
        json_d = json.load(f)
        img_name=json_d['imagePath'].split("\\")[-1].split(".")[0]+".jpg"
        img_path=os.path.join(img_folder_path,img_name)
        print("img_path:\t",img_path)
        relative_coordinate_txt(img_name,json_d,img_path)
        absolute_coordinate_txt(img_name,json_d,img_path)

5 从txt查看标注结果

import json
import os
import cv2
img_folder_path=r'F:\imagedata\FOV\FOV_SRC_2\imgs'#图片存放文件夹
folder_path=r"F:\imagedata\FOV\FOV_SRC_2\annotation"#标注数据的文件地址
txt_folder_path = r"F:\imagedata\FOV\FOV_SRC_2\txtlabel"#转换后的txt标签文件存放的文件夹


#相对坐标格式
def show_label_from_txt(img_path,txt_path):
    window_name = ('src') 
    cv2.namedWindow(window_name,cv2.WINDOW_FREERATIO)
    src_img=cv2.imread(img_path)    
    h,w = src_img.shape[:2]
    font = cv2.FONT_HERSHEY_SIMPLEX
    with open(temp_path, "r", encoding='utf-8') as f:
        lines = f.readlines()
    for line in lines:
        data = line.split(' ')
        label = data[1]
        x1 = int((float(data[3]) - float(data[7])/2)*w)
        y1 = int((float(data[5]) - float(data[9])/2)*h)
        x2 = int((float(data[3]) + float(data[7])/2)*w)
        y2 = int((float(data[5]) + float(data[9])/2)*h)      
        p1 = (x1,y1)
        p2 = (x2,y2)
        cv2.rectangle(src_img, p1, p2, (0, 250, 0), 2)
        cv2.putText(src_img,label,p1,font,2,(0,255,255),3)
        cv2.imshow(window_name,src_img)
        cv2.waitKey(0)
    cv2.destroyAllWindows()
    return 

i=0
for txtfile in os.listdir(txt_folder_path):
    temp_path=os.path.join(txt_folder_path,txtfile)
    
    i+=1
    if i>15:
        break
    #如果是一个子目录就继续
    if os.path.isdir(temp_path):
        continue
    print("txt_path:\t",temp_path)
    img_name=txtfile.split("\\")[-1].split(".")[0]+".jpg"
    img_path=os.path.join(img_folder_path,img_name)
    show_label_from_txt(img_path,temp_path)