voc获取文件夹下所有文件的文件名（去掉后缀）

import os

def get_file_names(folder_path = r'C:\Users\mage\Desktop\Timer\Segformer\datasets\ImageSets\JPEGImages', output_file_name = 'test.txt'):
    # 获取文件夹下所有文件的文件名（去掉后缀）
    file_names = [os.path.splitext(file)[0] for file in os.listdir(folder_path) if os.path.isfile(os.path.join(folder_path, file))]

    with open(output_file_name, 'w') as f:
        for name in file_names:
            f.write(name + '\n')