python批量读取指定目录下图片,并将其绝对路径写入txt文件

# coding: utf-8
import os

def createFilelist(images_path, text_save_path):
    # 打开图片列表清单txt文件
    file_name = open(text_save_path, "w")
    # 查看文件夹下的图片
    images_name = os.listdir(images_path)
    # 遍历所有文件
    for eachname in images_name:
        # 按照需要的格式写入目标txt文件
        file_name.write(images_path + '/' + eachname + '\n')
        
    print('生成txt成功!')

    file_name.close()



if __name__ == "__main__":
    # txt文件存放目录
    txt_path = '/home/alex/darknet-master/data' 
    # 图片存放目录
    images_path = '/home/alex/Desktop/deeplearning/python_exercise/pics'
    # 生成图片列表文件命名
    txt_name = 'test_detect.txt'
    if not os.path.exists(txt_name):
        os.mkdir(txt_name)
    # 生成图片列表文件的保存目录
    text_save_path = txt_path + '/' + txt_name
    #生成txt文件
    createFilelist(images_path, text_save_path)

参考文献:
https://blog.csdn.net/shirleycyy/article/details/79748873

此博客仅作学习记录,每天进步一点点

你可能感兴趣的:(python批量读取指定目录下图片,并将其绝对路径写入txt文件)