之前一直用windows下的bat脚本获取一个目录下的指定格式的文件名，如下所示：
dir *.jpg /b/s > train.set pause
十分简单，将这个bat文件放到你想要获取文件名的目录下，然后双击运行就可以将你想要保存的文件名写在train.set文件之中了。
今天由于实际需求，这个功能需要用python来实现，在网上查了一下发现，python脚本实现起来也很方便，而且也比较灵活，如下所示：
# -*- coding: utf-8 -*-
"""
Created on Mon Dec 12 14:59:46 2016 @author: shenruixue to calculate size after filter in conv and pool
"""
import os.path, time
import exceptions
import glob
import os class TypeError (Exception):
pass
if __name__ == '__main__':
file_srx = open("train.set")#其中包含所有待计算的文件名
file_object = open('all_time.txt', 'w')
line = file_srx.readline()
diff_time_all = 0
while line:
f = line[:-1] # 除去末尾的换行符
print (f)
print ('***********************************************************')
mtime = time.ctime(os.path.getmtime(f))
ctime = time.ctime(os.path.getctime(f))
mtime_s = (os.path.getmtime(f))
ctime_s = (os.path.getctime(f))
print "Last modified : %s, last created time: %s" % (mtime, ctime)
diff_time = (int(mtime_s) - int(ctime_s))
diff_time_all = diff_time_all + diff_time
print "diff time is ", diff_time f_record = f[:-17] #根据f给f_record进行赋值，需要根据实际情况修改17的大小
print (f_record)
print ('*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*')
os.chdir(f_record)
for file_name in glob.glob("*.jpg"):
print file_name
print ('*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*') file_object.write(file_name + " time is " + str(diff_time) + "\n")
line = file_srx.readline()
print "diff_time_all is ", diff_time_all file_object.write(str(diff_time_all))
file_object.close( )
可以看到用到了os与glog两个包，很方便快捷。
可以写成一个小脚本，以后方便使用。如下：
# -*- coding: utf-8 -*-
"""
Created on Mon Dec 12 14:59:46 2016 @author: shenruixue to calculate size after filter in conv and pool
"""
import os.path, time
import exceptions
import glob
import os
rootdir = os.getcwd()
if __name__ == '__main__':
file_object = open('name.set', 'w')
for file_name in glob.glob("*.jpg"):
print file_name
print ('*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*')
file_object.write(rootdir + file_name + "\n")
file_object.close( )