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

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

import os
import os.path
import sys
import glob
import shutil
def 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 txtfilenames

if __name__=="__main__": 
    txtpath="name.txt"
    filenames=getalltxtfilename2("/home/XXX/XXX,"name.txt")