参考：Python批量读取文件名并保存至txt_废话少说99的博客_博客
我只需要把所有图片的文件名保存在创建的列表中即可，所以将参考博主的代码稍加修改了一下，仅供自己学习记录
代码如下：
# P01 批量读取文件名，并将读取的文件名保存到指定路径下的txt中（带.*** 后缀） # 参考： https://blog.csdn.net/GeekYao/article/details/105074574 import os dir = "E:\\dBeker_copy\\tf-faster-rcnn-vgg16\\data\\demo\\" # 读取文件路径 im_names = [] wildcard = ".jpg" # 读取jpg图片 exts = wildcard.split(" ") files = os.listdir(dir) for name in files: fullname = os.path.join(dir, name) for ext in exts: if (name.endswith(ext)): #file.write(name + "\n") #print(name) # 文件名，如000001.jpg im_names.append(name) break print(im_names)最终我的显示结果为： 总共有300张图片，截取了一部分
非常感谢博主 @废话少说99 分享的干货~~~
