python输出文件的位置信息

记录如何用python输出文件的位置信息

1.要读取的文件的绝对位置：

path='/home/xxx/yolov5/runs/detect/exp77/' # 文件路径

2.指定要被识别的文件格式

if os.path.splitext(filename)[1]=='.jpg': # 写入的文件后缀

3.输出格式

out_path='data/obj_train_data/'+filename #输出特殊命名/相对位置

#out_path= path + filename #输出绝对位置

4.自动生成一个含有图片位置信息的train.txt文件，位于被测图片同级目录下

jpg_file = path + 'train.txt'

完整代码如下：

import os

def file_write_jpg(path,jpg_file):
   f = open(jpg_file,'w')
   filenames=os.listdir(path)
   for filename in filenames:
    if os.path.splitext(filename)[1]=='.jpg':  # 写入的文件后缀
        out_path='data/obj_train_data/'+filename #输出特殊命名/相对位置
        #out_path= path + filename #输出绝对位置
        print(out_path)
        # f.writable(out_path+'\n')
        f.write(out_path+'\n')
   f.close()

path='/home/xxx/yolov5/runs/detect/exp77/' # 文件路径
jpg_file = path + 'train.txt'
file_write_jpg(path,jpg_file)

终端输出效果：