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

ygdwxx

于 2020-08-17 15:34:29 发布

阅读量1.7k

点赞数 2

分类专栏： # Python编程

本文链接：https://blog.youkuaiyun.com/yangguidewxx/article/details/108057042

版权

Python编程专栏收录该内容

14 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -- 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()