Python读取指定路径所有.png (.txt等) 文件，将文件名写入.txt

最新推荐文章于 2023-07-15 09:53:24 发布

原创最新推荐文章于 2023-07-15 09:53:24 发布 · 1.7k 阅读

5 ·

CC 4.0 BY-SA版权

Python编程专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种使用Python脚本来生成用于目标检测训练的文件列表的方法。通过遍历指定目录下的所有.png文件，并将文件名写入trainval.txt中，为3D目标检测算法预研中的3DSSD项目准备数据集。

# -- coding: utf-8 --
# 生成trainval.txt文件
import os  

def file_name(file_dir):   
    L=[]   
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if os.path.splitext(file)[1] == '.png':  
                # L.append(os.path.join(root, file))  
                file_name = file[0:-4]+'.png' #去掉.txt
                L.append(file_name)  
    return L  
    
label_folder = '/home/yangguide/Desktop/work/工作/项目工程代码/自动驾驶/3D目标检测算法预研/3DSSD/tf14/3DSSD/dataset/KITTI/object/training/image_2'
trainval_file = '/home/yangguide/Desktop/work/工作/项目工程代码/自动驾驶/3D目标检测算法预研/3DSSD/tf14/3DSSD/dataset/KITTI/object/val.txt'

txt_name = file_name(label_folder)

with open(trainval_file, 'w') as f:
  for i in txt_name:
    f.write('{}\n'.format(i))
f.close()