python脚本制作dlib数据集

前提:你已经得到了预选框范围,来自动制作dlib数据集(xml),从而省去imglab花框过程
预选框文件内容格式(图片名,左上角x,左上角y,右下角x,右下角y),如:left_88.png 294 170 115 190
python脚本:

root_path="'/home/ubuntu/sotfware/dlib-18.16-new/examples/tiaotong/"
k=open('a.xml','w')
k.write("")
k.write("\n")
k.write("")
k.write("\n")
k.write("")
k.write("\n")
k.write("imglab dataset")
k.write("\n")
k.write("Created by imglab tool.")
k.write("\n")
k.write("")
k.write("\n")
f=open('11.txt','r')
c=[]
c.append('0')
for line in open('11.txt'):
    #line=f.readline()
    filename=line.split(' ')[0]
    left_top_x=line.split(' ')[2]
    left_top_y=line.split(' ')[1]
    width=line.split(' ')[3]
    height=line.split(' ')[4].strip()
    #print height
    if c[-1]==filename:
        k.write("    ".format(left_top_x,left_top_y,width,height))
        k.write("\n")
    else:
        if len(c)>1:
            k.write("  ")
            k.write("\n")
        k.write("  ".format(root_path,filename))
        k.write("\n")
        k.write("    ".format(left_top_x,left_top_y,width,height))
        k.write("\n")
    c.append(filename)
f.close()
k.write("  ")
k.write("\n")
k.write("")
k.write("\n")
k.write("")
k.close()

路径自行修改,执行该脚本,即可得到a.xml

你可能感兴趣的:(python,demo,python,脚本,数据)