python方法生成txt标签文件的实例代码


Posted in Python onMay 10, 2018

1、如果想要利用代码(不论是python、c++、亦或是matlab)实现生成标签文件,首先,也是灰常重要的一件事就是你的图片集一定要是有规律的命名。数字字母顺序排开。这一点非常重要,相关重命名方法请自行百度或请教大牛。

2、如图为博主的图片集。(注意命名规律)

博主只分为猫和鸟两类。

python方法生成txt标签文件的实例代码

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 相关文章推荐
Python Web服务器Tornado使用小结
May 06 Python
Python中的super用法详解
May 28 Python
Python实现注册登录系统
Aug 08 Python
Python方法的延迟加载的示例代码
Dec 18 Python
Python数据结构与算法(几种排序)小结
Jun 22 Python
Django中create和save方法的不同
Aug 13 Python
python支持多线程的爬虫实例
Dec 21 Python
python实现可下载音乐的音乐播放器
Feb 25 Python
Python爬取365好书中小说代码实例
Feb 28 Python
Pycharm 使用 Pipenv 新建的虚拟环境(图文详解)
Apr 16 Python
Python matplotlib模块及柱状图用法解析
Aug 10 Python
Python实现信息管理系统
Jun 05 Python
Python处理菜单消息操作示例【基于win32ui模块】
May 09 #Python
Python创建普通菜单示例【基于win32ui模块】
May 09 #Python
Python使用Windows API创建窗口示例【基于win32gui模块】
May 09 #Python
Python基于win32ui模块创建弹出式菜单示例
May 09 #Python
numpy中的delete删除数组整行和整列的实例
May 09 #Python
java中两个byte数组实现合并的示例
May 09 #Python
Python 使用PIL numpy 实现拼接图片的示例
May 08 #Python
You might like
PHP令牌 Token改进版
2008/07/18 PHP
php下连接ftp实现文件的上传、下载、删除文件实例代码
2010/06/03 PHP
smarty自定义函数用法示例
2016/05/20 PHP
Yii框架实现对数据库的CURD操作示例
2019/09/03 PHP
Ajax::prototype 源码解读
2007/01/22 Javascript
给Javascript数组插入一条记录的代码
2007/08/30 Javascript
基于JQuery模仿苹果桌面的Dock效果(初级版)
2012/10/15 Javascript
A标签中通过href和onclick传递的this对象实现思路
2013/04/19 Javascript
以JSON形式将JS中Array对象数组传至后台的方法
2014/01/06 Javascript
javascript 获取函数形参个数
2014/07/31 Javascript
Javascript 拖拽雏形中的一些问题(逐行分析代码,让你轻松了拖拽的原理)
2015/01/23 Javascript
jquery获取url参数及url加参数的方法
2015/10/26 Javascript
javascript显示倒计时控制按钮的简单实现
2016/06/07 Javascript
JavaScript随机打乱数组顺序之随机洗牌算法
2016/08/02 Javascript
JavaScript 上传文件(psd,压缩包等),图片,视频的实现方法
2017/06/19 Javascript
vue.js根据代码运行环境选择baseurl的方法
2018/02/28 Javascript
Node.js一行代码实现静态文件服务器的方法步骤
2019/05/07 Javascript
vue图片加载失败时用默认图片替换的方法
2019/08/29 Javascript
three.js中多线程的使用及性能测试详解
2021/01/07 Javascript
举例讲解Python的Tornado框架实现数据可视化的教程
2015/05/02 Python
Python制作钉钉加密/解密工具
2016/12/07 Python
python简易远程控制单线程版
2018/06/20 Python
python之super的使用小结
2018/08/13 Python
Python高级特性与几种函数的讲解
2019/03/08 Python
Django中使用极验Geetest滑动验证码过程解析
2019/07/31 Python
python-web根据元素属性进行定位的方法
2019/12/13 Python
html5中使用hotcss.js实现手机端自适配的方法
2020/04/23 HTML / CSS
美国创意礼品网站:UncommonGoods
2017/02/03 全球购物
T3官网:头发造型工具
2019/12/26 全球购物
语文复习计划
2015/01/19 职场文书
行政司机岗位职责
2015/04/10 职场文书
召开会议通知范文
2015/04/15 职场文书
2015年新农村建设工作总结
2015/05/22 职场文书
挂职锻炼工作总结2015
2015/05/28 职场文书
2016年世界艾滋病日宣传活动总结
2016/04/01 职场文书
《西游记》读后感(3篇)
2019/09/20 职场文书