python批量读取图片生成保存图片路径的txt文档

---

前言

我们做深度学习项目时，训练时会通过读取图片数据集存放的路径来获取数据信息。这时，我们就需要通过python脚本程序去生成保存有图片存放路径的txt文件，当然保存有图片存放路径的txt文件有很多用处，作者在做海思3516dv300开发板和瑞芯微rv1109开发板算法移植时，量化模型需要准备一定数量训练集图片做量化，这时候就需要该脚本。

---

一、完整代码如下

import os
import sys
def listfiles(rootDir, txtfile, label=0):
    ftxtfile = open(txtfile, 'w')
    list_dirs = os.walk(rootDir)
    count = 0
    dircount = 0
    #遍历文件夹中图片，并读取路径信息
    for root,dirs,files in list_dirs:	
       for d in dirs:
           print(os.path.join(root, d))
           dircount += 1
       for f in files:
           print(os.path.join(root, f))
           ftxtfile.write(os.path.join(root, f)+' ' + '\n')
           count += 1
       print(rootDir + ' has ' + str(count) + ' files')		#打印出txt中每行路径
if __name__ == '__main__':
	#'E:\Excise\dataset\data_Evehicle\image_towk'为图片存放路径，文件夹名字为image_towk
	#'E:\Excise\dataset\data_Evehicle\image_towk\test.txt' 生成的图片路径会存放到test.txt
  listfiles(r'E:\Excise\dataset\data_Evehicle\image_towk', r'E:\Excise\dataset\data_Evehicle\image_towk\test.txt')

二、运行结果

总结

以上就是今天要讲的内容，脚本超级好用，学好python，大大提高工作效率！觉得还不错的，感谢关注收藏，后续还会继续分享好用的数据处理脚本。