记录如何用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)
终端输出效果: