伯克利发布BDD100K标签格式转成YOLOV3的txt格式

import json
categorys=['car','truck', 'bus']
jsonFile="./bdd100k_labels_images_train.json"

f=open(jsonFile)
info=json.load(f)

for image_index in range(0,len(info)):
    strs=""
    image=info[image_index]
    write=open("./train/%s.txt"%(image["name"][0:-4]),'w')  #需先建立train文件夹
    for labels_index in range(0,len(image["labels"])):
        
        labels=image["labels"][labels_index]
        if labels["category"] in categorys:
            dw = 1.0/1280
            dh = 1.0/720
            strs+=str(categorys.index(labels["category"]))
            strs+=" "
            strs+=str(((labels["box2d"]["x1"] + labels["box2d"]["x2"])/2.0 - 1) * dw)
            strs+=" "
            strs+=str(((labels["box2d"]["y1"] + labels["box2d"]["y2"])/2.0 - 1) * dh)
            strs+=" "
            strs+=str(((labels["box2d"]["x2"] - labels["box2d"]["x1"])) * dw)
            strs+=" "
            strs+=str(((labels["box2d"]["y2"] - labels["box2d"]["y1"])) * dh) 
            strs+="\n"
    write.writelines(strs)
    write.close()
    print("%s has been dealt!"%image["name"])

训练部分样本