import osdef 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')