python方法生成txt标签文件的实例代码
2018-05-10 08:43
1041 查看
1、如果想要利用代码(不论是python、c++、亦或是matlab)实现生成标签文件,首先,也是灰常重要的一件事就是你的图片集一定要是有规律的命名。数字字母顺序排开。这一点非常重要,相关重命名方法请自行百度或请教大牛。
2、如图为博主的图片集。(注意命名规律)
博主只分为猫和鸟两类。
3、看代码。
在caffe根目录下创建一个我们的工程目录my-caffe-project
创建并编辑create_db.py文件,使用如下指令:
vim create_db.py
然后,代码内容就是我们的重点了。注意相关变量和路径的修改。
# -*- coding: UTF-8 -*- import os import re """ 函数说明:生成图片列表清单txt文件 Parameters: images_path - 图片存放目录 txt_save_path - 图片列表清单txt文件的保存目录 Returns: 无 Author: Jack Cui Modify: 2017-03-29 """ def createFileList(images_path, txt_save_path): #打开图片列表清单txt文件 fw = open(txt_save_path,"w") #查看图片目录下的文件,相当于shell指令ls images_name = os.listdir(images_path) #遍历所有文件名 for eachname in images_name: #正则表达式这里可以根据情况进行更改,如果多类记得仿照下面定义相关变量 #正则表达式规则:找以cat开头,紧跟0到49个数字,并以jpg结尾的图片文件 pattern_cat = r'(^cat\d{0,49}.jpg$)' #正则表达式规则:找以bird开头,紧跟0到49个数字,以jpg结尾的图片文件 pattern_bird = r'(^bird\d{0,49}.jpg$)'#更改变量名需要定义 #正则表达式匹配 cat_name = re.search(pattern_cat, eachname) bird_name = re.search(pattern_bird, eachname) #按照规则将内容写入txt文件中 if cat_name != None: fw.write(cat_name.group(0) + ' 1\n') if bird_name != None: fw.write(bird_name.group(0) + ' 2\n')#如果分为多类,多加几个if便是。当然相关变量也要记得定义。 #打印成功信息 print "生成txt文件成功" #关闭fw fw.close() #下面是相关变量定义的路径 if __name__ == '__main__': #caffe_root目录 caffe_root = '/home/xn/caffe/' #my-caffe-project目录 my_caffe_project = caffe_root + 'my-caffe-project/' #图片存放目录 images_path = caffe_root + 'data/mydata/myimagenet/train/' #生成的图片列表清单txt文件名 txt_name = 'train.txt' #生成的图片列表清单txt文件的保存目录 txt_save_path = my_caffe_project + txt_name #生成txt文件 createFileList(images_path, txt_save_path)
完成。如果想要生成测试标签文件类似。
以上这篇python方法生成txt标签文件的实例代码就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章推荐
- python方法生成txt标签文件代码
- python实现对文件中图片生成带标签的txt文件方法
- php 批量生成html,txt文件的方法(实例代码)
- 详解python脚本自动生成需要文件实例代码
- python实现对文件中图片生成带标签的txt文件
- 用python生成1000个txt文件的方法
- php生成txt文件实例代码介绍
- php生成txt文件实例代码介绍
- Python三种遍历文件目录的方法实例代码
- python生成以及打开json、csv和txt文件的实例
- C#读写txt文件多种方法实例代码
- python实例31[列出目录下所有的文件到txt]
- python生成大文件 100G瞬间生成 三种方法
- eclipse develop for C\C++ 文件注释函数注释代码模板自动生成方法
- python 生成目录树及显示文件大小的代码
- php获取远程文件大小的三种方法(实例代码)
- delphi 生成xml 方法 与 delphi txt文件操作
- php代码生成txt文件并下载
- 使用VBS访问外部文本文件一些方法和脚本实例代码
- webservice(六)基于契约优先的wsdl文件生成webservice服务代码实例