python3批量读取图片名（2020.10.08）

参考：Python批量读取文件名并保存至txt_废话少说99的博客_CSDN博客

我只需要把所有图片的文件名保存在创建的列表中即可，所以将参考博主的代码稍加修改了一下，仅供自己学习记录

代码如下：

# 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 分享的干货~~~