把某个文件夹下的图片名称写入txt文件

最新推荐文章于 2024-07-18 22:57:16 发布

原创最新推荐文章于 2024-07-18 22:57:16 发布 · 1.8k 阅读

3 ·

CC 4.0 BY-SA版权

python 同时被 2 个专栏收录

3 篇文章

订阅专栏

utils

1 篇文章

订阅专栏

本文介绍了一个简单的Python脚本，该脚本可以遍历指定目录下的所有.bmp文件，并将这些文件的名字（不含扩展名）写入到一个文本文件中。此外，还提供了一个不考虑文件扩展名的通用文件列表生成方法。

import os


def ListFilesToTxt(input_dir, file, wildcard, recursion):
    exts = wildcard.split(" ")
    for root, subdirs, files in os.walk(input_dir):
        for name in files:
            for ext in exts:
                if (name.endswith(ext)):
                    file.write(name + "\n")
                    break
        if (not recursion):
            break


def Test():
    input_dir = "pedestrian_detection/dataset/BMPImages"
    outfile = "train_list.txt"
    # wildcard = ".txt .exe .dll .lib"
    wildcard = ".bmp"
    file = open(outfile, "w")
    if not file:
        print("cannot open the file %s for writing" % outfile)

    ListFilesToTxt(input_dir, file, wildcard, 0)

    file.close()


Test()

不带文件后缀

import os

names = os.listdir('pedestrian_detection/dataset/BMPImages')  # 路径
i = 0  # 用于统计文件数量是否正确，不会写到文件里
train_val = open('test.txt', 'w')
for name in names:
    index = name.rfind('.')
    name = name[:index]
    train_val.write(name + '\n')
    i = i + 1

print(i)