博客

前言：

学习过程中经常需要进行数据处理，其中最常见的操作就是读取某个文件夹中的所有文件名称，并将这些文件名称按照读取顺序一行一行写入txt文本文件中。记录一下

一.核心代码

1. os.listdir(filePath) # os.listdir方法返回一个列表对象，参数为
   
   文件夹的路径，可以是”相对路径“和”绝对路径“
   
   2.在向txt文本文件中写入信息时，有时需要换行写入
   
   ==rowInfo = imageDir + ’ ’ + annotationDir == 不会换行写入
   
   ==rowInfo = imageDir + ’ ’ + annotationDir + ‘\n’ == 加上‘\n’,换行写入

import os
# readInfo函数，根据文件夹路径读取文件夹下所有文件名
def readInfo():
    filePath = 'F:\\ai\\PaddleSeg-release-2.4\\data\\island\\JPEGImages\\'
    name = os.listdir(filePath)         # os.listdir方法返回一个列表对象
    return name

# 程序入口
if __name__ == "__main__":
    fileList = readInfo()       # 读取文件夹下所有的文件名，返回一个列表
    print(fileList)
    file = open('train.txt', 'w')   # 创建文件，权限为写入
    for i in fileList:
        imageDir = 'JPEGImages/' + i
        annotationDir = 'Annotations/' + i
        rowInfo = imageDir + ' ' + annotationDir + '\n'
        print(rowInfo)
        file.write(rowInfo)