首先给出代码：
import os
path = "/home/data//"
path_exp = os.path.expanduser(path)
classes = [int(p) for p in os.listdir(path_exp)]
classes.sort()
# nrof_classes一个数据集下有多少个文件夹,就是说有多少个人,多少个类别
nrof_classes = len(classes)
count=0
files = open("train_list.txt",'w')
filets = open("test_list.txt",'w')
count_u=0
for i in range(nrof_classes):
class_name = str(classes[i])
count=count+1
count_u=count_u+1
facedir = os.path.join(path_exp, class_name)
prefix1 = path+class_name+"/"
if os.path.isdir(facedir):
images = os.listdir(facedir)
#print(images[0])
image_paths = [(prefix1+img+" "+class_name+"\n") for img in images]
#print(image_paths[0])
if count < 0.8*nrof_classes:
if len(image_paths)>4:
test_path=[]
for x in range(2):
test_path.append(image_paths[0])
del image_paths[0]
filets.writelines(test_path)
files.writelines(image_paths)
#if count==2:
# break
#imgae_pathses = []
#防止图像大小为0
#for x in image_paths:
# if os.path.getsize(x)>0:
# imgae_pathses.append(x)
#if len(imgae_pathses)==0:
# os.rmdir(facedir)
files.close()
filets.close() python下os模块的一下有用的用法：
0 重命名：文件和文件夹都是一个命令：
os.rename(original_dir,new_dir)
1 文件操作：
os.mknod("test.txt") 创建空文件
fp = open("test.txt",w) 直接打开一个文件，如果文件不存在则创建文件
关于open 模式：
w 以写方式打开，
a 以追加模式打开 (从 EOF 开始, 必要时创建新文件)
r+ 以读写模式打开
w+ 以读写模式打开 (参见 w )
a+ 以读写模式打开 (参见 a )
rb 以二进制读模式打开
wb 以二进制写模式打开 (参见 w )
ab 以二进制追加模式打开 (参见 a )
rb+ 以二进制读写模式打开 (参见 r+ )
wb+ 以二进制读写模式打开 (参见 w+ )
ab+ 以二进制读写模式打开 (参见 a+ )
fp.read([size]) #size为读取的长度，以byte为单位
fp.readline([size]) #读一行，如果定义了size，有可能返回的只是一行的一部分
fp.readlines([size]) #把文件每一行作为一个list的一个成员，并返回这个list。其实它的内部是通过循环调用readline()来实现的。如果提供size参数，size是表示读取内容的总长，也就是说可能只读到文件的一部分。
fp.write(str) #把str写到文件中，write()并不会在str后加上一个换行符
fp.writelines(seq) #把seq的内容全部写到文件中(多行一次性写入)。这个函数也只是忠实地写入，不会在每行后面加上任何东西。
fp.close() #关闭文件。python会在一个文件不用后自动关闭文件，不过这一功能没有保证，最好还是养成自己关闭的习惯。 如果一个文件在关闭后还对其进行操作会产生ValueError
fp.flush() #把缓冲区的内容写入硬盘
fp.fileno() #返回一个长整型的”文件标签“
fp.isatty() #文件是否是一个终端设备文件（unix系统中的）
fp.tell() #返回文件操作标记的当前位置，以文件的开头为原点
fp.next() #返回下一行，并将文件操作标记位移到下一行。把一个file用于for … in file这样的语句时，就是调用next()函数来实现遍历的。
fp.seek(offset[,whence]) #将文件打操作标记移到offset的位置。这个offset一般是相对于文件的开头来计算的，一般为正数。但如果提供了whence参数就不一定了，whence可以为0表示从头开始计算，1表示以当前位置为原点计算。2表示以文件末尾为原点进行计算。需要注意，如果文件以a或a+的模式打开，每次进行写操作时，文件操作标记会自动返回到文件末尾。
fp.truncate([size]) #把文件裁成规定的大小，默认的是裁到当前文件操作标记的位置。如果size比文件的大小还要大，依据系统的不同可能是不改变文件，也可能是用0把文件补到相应的大小，也可能是以一些随机的内容加上去。
也可以直接打开式的新建文件：
with open('path0.txt','ab') as f:
for d in arr:
np.savetxt(f,d,fmt='%5f')
以上就是python脚本生成caffe train_list.txt的方法的详细内容，更多请关注Gxl网其它相关文章！
查看更多关于python脚本生成caffetrain_list.txt的方法的详细内容...