voc数据集---截取图片名python

来源：互联网发布：反三角函数计算器软件编辑：程序博客网时间：2024/04/29 03:33

最近在搞深度学习，本来按自己的数据集每张数据的标记保存为txt，txt中第一行为标记，第二行为坐标点坐标，将txt转换为xml用的网上现成代码，然后现在实现train.txt，实现提取图片文件名的方法，matlab不打算用了，搞了一天c++，生成一个可执行文件，也是醉醉的。花了一个上午，写了一个python的

import osimport os.pathimport sysimport globimport shutildef getalltxtfilename2(path,txtpath):    txtfilenames=[]    txt=open(txtpath,'a')    for dirpath,dirnames,filenames in os.walk(path):        filesnames=filter(lambda filename:filename[-4:]==".jpg",filenames);        print("dir",dirpath)   # for dirp in filenames:        for filename in filenames:       # print os.path.join(dirpath,filename)            imgname=filename.split(".")[0]            txtfilenames.extend(filenames)            txt.write(imgname)            txt.write("\n")    txt.close()    return txtfilenamesif __name__=="__main__":     txtpath="name.txt"    filenames=getalltxtfilename2("/home/XXX/XXX,"name.txt")

0 0