python——将指定文件夹中所有JPG格式的文件路径及文件名按行写入txt文件中

机器视觉中，经常需要批量测试训练好的模型，有的测试文件只支持单张图像的测试，我们想要将大量图片进行批量测试并保存，有时就需要把图像文件的路径及文件名写在一个txt文件里，这里以JPG格式的图像为例，我的待处理文件夹如下：

代码如下：

# -*- coding: utf-8 -*-   
import time     
import os  
import shutil


def readFilename(path, allfile):
    filelist = os.listdir(path)

    for filename in filelist:
        filepath = os.path.join(path, filename)
        if os.path.isdir(filepath):
            readFilename(filepath, allfile)
        else:
            allfile.append(filepath)
    return allfile



if __name__ == '__main__':
    path1="/Users/sunny/Desktop/789/"
    allfile1=[]
    allfile1=readFilename(path1,allfile1)
    allname1=[]
    txtpath="/Users/sunny/Desktop/"+"JPGlist.txt"
    for name in allfile1:
        print name
        file_cls=name.split("/")[-1].split(".")[-1]
        if file_cls=='jpg':
            print name.split("/")[-1]
            with open(txtpath,'a+') as fp:
                fp.write("".join(name)+"\n")

运行结果如下：