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"])
伯克利发布BDD100K标签格式转成YOLOV3的txt格式
猜你喜欢
转载自blog.csdn.net/qq_41979513/article/details/100565782
今日推荐
周排行