Windows下实现将Pascal VOC转化为TFRecords


Posted in Python onFebruary 17, 2020

前言

由于TensorFlow常用TFrecords作为输入格式,我们需要将制作好的Pascal VOC转为TFrecords格式。使用 object detection API 内提供的脚本。

TFRecords格式如下:

-uint64 length

-uint32 masked_crc32_of_length

-byte data[length]

-uint32 masked_crc32_of_data

前置要求:

Anaconda3(使用python3.x)

制作好的Pascal VOC数据集

1、配置protobuf

Google的protobuf是一种轻便高效的结构化数据存储格式,结构扩展性高,速度快,密度大。我们下载对应的protoc-3.4.0-win32.zip版本。解压之后的文件内容如下:

Windows下实现将Pascal VOC转化为TFRecords

将bin文件夹所在路径添加到环境变量PATH,再将bin下的protoc.exe移动到C:\Windows\System32目录下。

cmd下进入~models/reseach目录下,输入:

protoc object_detection/protos/*.proto --python_out=.

2、设置搜索路径

Windows下的API使用时会出现图示bug,需要创建xx.pth文件,将以下两条路径添加进去:

~\models\research

~\models\research\slim

接着,将pth文件移动到python安装目录的site-packages下。

Windows下实现将Pascal VOC转化为TFRecords

最后,我们利用cmd运行model_builder_test.py文件,测试环境是否搭建成功。

Windows下实现将Pascal VOC转化为TFRecords

3、修改pascal_label_map.pbtxt

打开~models/research/object_detection/data/pascal_label_map.pbtxt,修改其中的name为自己的分类名。

如图所示,这里我要识别分类的是图片数字0~3,所以修改了对应的name。

Windows下实现将Pascal VOC转化为TFRecords

4、修改create_pascal_tf_record.py文件

(1)47行:根据自己的VOC数据集格式,修改年份,如果是2007,则保持默认;

(2)49行:将data/pascal_label_map.pbtxt改为绝对路径

(3)81行:改为img_path = os.path.join('JPEGImages', data['filename'])

(4)82行:改为full_path =os.path.join(dataset_directory, 'VOC2012',img_path+'.jpg')

(5)163行 : 改为examples_path =os.path.join(data_dir, year, 'ImageSets', 'Main','xx' + FLAGES.set + '.txt')

xx里的内容是~VOCdevkit\VOC2012\ImageSets\Main目录下的文件前缀名称,如果没有,留空就可以。

5、运行

cmd下输入:

python ~/models/research/object_detection/dataset_tools/create_pascal_tf_record.py\

--data_dir=~/VOCdevkit \

--year=VOC2012 \

--output_path=~/xx.record

其中,波浪线的地方分别为自己的py文件位置、VOC数据位置、转换的record文件存储位置。

注:

1、protoc必须使用3.4版本而非3.5,否则会出现找不到*.proto的bug;

2、运行时出现很多bug,根据bug一步一步回推代码修改。

以上这篇Windows下实现将Pascal VOC转化为TFRecords就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
用python实现面向对像的ASP程序实例
Nov 10 Python
Python中用format函数格式化字符串的用法
Apr 08 Python
书单|人生苦短,你还不用python!
Dec 29 Python
Python实现朴素贝叶斯分类器的方法详解
Jul 04 Python
使用Python-OpenCV向图片添加噪声的实现(高斯噪声、椒盐噪声)
May 28 Python
Python3.7安装keras和TensorFlow的教程图解
Jun 18 Python
numpy:np.newaxis 实现将行向量转换成列向量
Nov 30 Python
PHP基于phpqrcode类库生成二维码过程解析
May 28 Python
python3.7调试的实例方法
Jul 21 Python
python用tkinter实现一个简易能进行随机点名的界面
Sep 27 Python
Pycharm安装python库的方法
Nov 24 Python
Python实现滑雪小游戏
Sep 25 Python
tensorflow生成多个tfrecord文件实例
Feb 17 #Python
tensorflow将图片保存为tfrecord和tfrecord的读取方式
Feb 17 #Python
Python 读取有公式cell的结果内容实例方法
Feb 17 #Python
Python破解BiliBili滑块验证码的思路详解(完美避开人机识别)
Feb 17 #Python
Tensorflow 实现将图像与标签数据转化为tfRecord文件
Feb 17 #Python
将自己的数据集制作成TFRecord格式教程
Feb 17 #Python
tensorflow 实现数据类型转换
Feb 17 #Python
You might like
输出控制类
2006/10/09 PHP
PHP生成sitemap.xml地图函数
2013/11/13 PHP
php断点续传之如何分割合并文件
2014/03/22 PHP
ThinkPHP模板判断输出Defined标签用法详解
2014/06/30 PHP
CodeIgniter表单验证方法实例详解
2016/03/03 PHP
php集成动态口令认证
2016/07/21 PHP
PHP封装XML和JSON格式数据接口操作示例
2019/03/06 PHP
jQuery获取Select选择的Text和Value(详细汇总)
2013/01/25 Javascript
javascript动态修改Li节点值的方法
2015/01/20 Javascript
JavaScript中字符串(string)转json的2种方法
2015/06/25 Javascript
js文本框输入内容智能提示效果
2015/12/02 Javascript
javascript与jquery中的this关键字用法实例分析
2015/12/24 Javascript
JavaScript自动点击链接 防止绕过浏览器访问的方法
2017/01/19 Javascript
Angular.JS中指令ng-if、ng-show/ng-hide和ng-switch的使用教程
2017/05/07 Javascript
AngularJS 限定$scope的范围实例详解
2017/06/23 Javascript
Vue实现virtual-dom的原理简析
2017/07/10 Javascript
JavaScript碎片—函数闭包(模拟面向对象)
2019/03/13 Javascript
小程序多图列表实现性能优化的方法步骤
2019/05/28 Javascript
[14:57]DOTA2 HEROS教学视频教你分分钟做大人-幽鬼
2014/06/13 DOTA
Python的lambda匿名函数的简单介绍
2013/04/25 Python
在RedHat系Linux上部署Python的Celery框架的教程
2015/04/07 Python
介绍Python的@property装饰器的用法
2015/04/28 Python
Python基于identicon库创建类似Github上用的头像功能
2017/09/25 Python
Python切片操作实例分析
2018/03/16 Python
tensorflow 1.0用CNN进行图像分类
2018/04/15 Python
Python爬虫框架Scrapy基本用法入门教程
2018/07/26 Python
python中通过pip安装库文件时出现“EnvironmentError: [WinError 5] 拒绝访问”的问题及解决方案
2020/08/11 Python
HTML5 File接口在web页面上使用文件下载
2017/02/27 HTML / CSS
使用phonegap查找联系人的实现方法
2017/03/31 HTML / CSS
英国口碑最好的的维他命胶囊品牌:Myvitamins(有中文站)
2016/12/03 全球购物
英国在线珠宝店:The Jewel Hut
2017/03/20 全球购物
泰国在线书店:SE-ED
2020/06/21 全球购物
项目开发计划书
2014/01/09 职场文书
中层干部竞争上岗演讲稿
2014/01/13 职场文书
人事任命书范文
2014/06/04 职场文书
交警失职检讨书
2015/01/26 职场文书