MyTool--将文件夹里面的所有文件名记录在txt中

方法一：我的

version2

import os

# 图片所在的文件夹
folder_path = './mask'
# 将文件名存储在列表里
list_jpg = os.listdir(folder_path)
# 排序
list_jpg = sorted(list_jpg)

if __name__ == '__main__':
    '''
    1 要在此文件目录下事先创建test.txt文件
    2 写入txt文件
    3 与后面完成train.txt分开单独使用时, 直接注释掉另一个
    '''
    with open('test.txt', 'w') as f:  # train.txt或者test.txt
        for i in list_jpg:
            # 如果去掉图片后缀
            i = i.split(".")[0]  # xxx.png -> xxx

            f.writelines(i + '\n')

version1 

import os

filePath = './mask'
list_jpg = os.listdir(filePath)
list_jpg = sorted(list_jpg)
if __name__ == '__main__':
    '''
    => 用于完成test.txt, 要在此文件目录下事先创建test.txt文件
    与后面完成train.txt分开单独使用时, 直接注释掉另一个
    '''
    # /=== ===> test.txt 写入pic_name & label <=== ===\
    with open('test.txt', 'w') as f:  # train.txt或者test.txt
        for i in list_jpg:
            # 如果去掉图片后缀
            i = i.split(".")[0]
            
            f.writelines(i + '\n')

    # \=== ===> test.txt 写入pic_name & label <=== ===/

 

方法二：网友的