python将一个文件夹下图片保存在txt文本

原创于 2018-08-03 16:19:08 发布 · 1.9k 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个简单的Python脚本，该脚本可以遍历指定文件夹内的所有图片，并将图片名称（不包含扩展名）保存到一个TXT文件中。此方法适用于自动化整理大量图片文件名的场景。

#!/usr/bin/env python 
# -*- coding:utf-8 -*-
# 将一个文件夹下图片保存在txt文本
import sys
sys.path.append('D:\\train\\img\\image_valid') （图片文件夹）
import os
import random
img_dir = 'D:\\train\\img\\image_valid'
file_list = []
write_dir_name = 'D:\\train\\img\\img_valid.txt'  （图片文件名，不含jpg）
write_file = open(write_dir_name, "w")
for file in os.listdir(img_dir):
##    if file.endswith(".jpg"):
    write_name = file
    write_name1 = write_name.split('.')
    print(write_name1)
    write_name2 = write_name1[0]
    print(write_name2)
    file_list.append(write_name2)
    sorted(file_list)
num = len(file_list)
for current_line in range(num):
    write_file.write(file_list[current_line] + '\n')
write_file.close()