将图像数据读写标记成txt文件
2017-09-30 11:54
399 查看
# -*- coding: UTF-8 -*- import os def IsSubString(SubStrList,Str): flag=True for substr in SubStrList: if not(substr in Str): flag=False return flag #扫面文件 def GetFileList(FindPath,FlagStr=[]): FileList=[] FileNames=os.listdir(FindPath) if len(FileNames)>0: for fn in FileNames: if len(FlagStr)>0: if IsSubString(FlagStr,fn): fullfilename=os.path.join(FindPath,fn) FileList.append(fullfilename) else: fullfilename=os.path.join(FindPath,fn) FileList.append(fullfilename) if len(FileList)>0: FileList.sort() return FileList train_txt=open('train.txt','w') imgfile=GetFileList('/home/disk/lyw/AuoPose/DATA/images/train/carbody/')#将数据集放在与.py文件相同目录下 for img in imgfile: str1=img+' '+'1'+'\n' #用空格代替转义字符 \t train_txt.writelines(str1) imgfile=GetFileList('/home/disk/lyw/AuoPose/DATA/images/train/carhead/') for img in imgfile: str2=img+' '+'0'+'\n' train_txt.writelines(str2) #train_txt.close() imgfile=GetFileList('/home/disk/lyw/AuoPose/DATA/images/train/cartrail/') for img in imgfile: str3=img+' '+'2'+'\n' train_txt.writelines(str3) train_txt.close() #测试集文件列表 test_txt=open('val.txt','w') imgfile=GetFileList('/home/disk/lyw/AuoPose/DATA/images/val/carbody/')#将数据集放在与.py文件相同目录下 for img in imgfile: str4=img+' '+'1'+'\n' test_txt.writelines(str4) imgfile=GetFileList('/home/disk/lyw/AuoPose/DATA/images/val/carhead/') for img in imgfile: str5=img+' '+'0'+'\n' test_txt.writelines(str5) imgfile=GetFileList('/home/disk/lyw/AuoPose/DATA/images/val/cartrail/') for img in imgfile: str6=img+' '+'2'+'\n' test_txt.writelines(str6) test_txt.close() print("ok")
相关文章推荐
- OpenCV学习笔记 cv.Mat 与 .txt 文件数据的读写操作
- 黑白BMP文件的读写和数据黑白图像显示
- OpenCV学习笔记 cv.Mat 与 .txt 文件数据的读写操作
- C#读写txt文件数据的两种方法
- cv.Mat 与 .txt 文件数据的读写操作
- 数据压缩原理与应用 图像文件的读写和转换(BMP2YUV)实验报告
- C语言读取txt里面的图像数据字符串换成十六进制然后保存为bmp格式的图像文件
- OpenCV cv.Mat与.txt文件数据的读写操作
- OpenCV学习笔记 cv.Mat 与 .txt 文件数据的读写操作
- 数据压缩原理 实验二 图像文件的读写和转换
- 【MATLAB】导入包含数据的txt文件到MATLAB中,并绘制图像
- 【python图像处理】txt文件数据的读取与写入
- C++读写txt文件方式以及基于opencv的Mat数据类型读写txt文件类型
- opencv输出图像数据到txt文件
- [C++]读写txt文件和从txt文件中读取数据写入数组中
- OpenCV数字图像处理十:读写txt文件,在图像训练和测试的时候经常用到
- C++读写txt文件方式以及基于opencv的Mat数据类型读写txt文件
- MATLAB将图像数据以txt文件保存以及从txt文件中读矩阵并显示图像
- 【项目】Project数据读写小程序(二):Project文件xml格式的标记
- OpenCV学习笔记(23)cv.Mat 与 .txt 文件数据的读写操作