机器视觉中,经常需要批量测试训练好的模型,有的测试文件只支持单张图像的测试,我们想要将大量图片进行批量测试并保存,有时就需要把图像文件的路径及文件名写在一个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)
allname