tensorflow object detection API训练公开数据集Oxford-IIIT Pets Dataset
为了避免不必要的麻烦,先说一下我的系统版本
Python 3.6 tensorflow 1.10 windows 7
object detection API安装
object detection API 安装参见官方的github: https://github.com/tensorflow/models/tree/master/research/object_detection
这里需要特别强调一下,一定要检查一下,下面的python的包都安装了,方法很简单,全部执行一遍
coco API安装
windows下面不需要--user选项,Oxford-IIIT Pet 数据集使用coco metrix, 所以下面必须执行这个命令行:
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
在windows下面遇到utf-8编码错误
如下修正:
C:\Program Files\Python\Python36\Lib\site-packages\pip\compat\_init_.py
79行改成gbk, utf-8替换为gbk, 不一定是是79行。再次执行即可成功安装
创建训练数据记录tfrecord
下载好Oxford-IIIT Pets Dataset数据集,解压缩到这里
然后执行下面的命令行:
训练数据成功创建在指定目录:先切换到指定目录,完整的命令行执行
D:\tensorflow\models\research>python objectdetection/datasettools/createpettfrecord.py --labelmappath=objectdetection/data/petlabelmap.pbtxt
--datadir=D:/petdata --output-dir=D:\tensorflow\mytrain\data
还不会看这里:
https://github.com/tensorflow/models/blob/master/research/objectdetection/g3doc/preparing_inputs.md
迁移学习
这步成功以后,就可以开始执行真正的训练啦,等等,别着急,我们是基于预训练模型的迁移学习,所以还有几件事情必须搞定,
下载预训练的tensorflow模型,我这里下载的是
http://download.tensorflow.org/models/objectdetection/ssdmobilenetv1coco201801_28.tar.gz
解压缩到指定目录,我的完整目录结构如下:
其实有+加号的表示目录文件夹, - 表示文件
labelmap file来自 D:\tensorflow\models\research\objectdetection\data\petlabelmap.pbtxt pipeline config file来自 D:\tensorflow\models\research\objectdetection\samples\configs\ssdmobilenetv1pets.config
直接copy过来,然后打开
开始修改,把所有【PATH_TO_BE_CONFIGURED】都改到正确路径上来:
保存好啦,然后直接执行训练的命令行:
各个参数选项解释如下:
--pipelineconfigpath
训练时候配置目录,所有关于训练各种输入路径、参数模型、参数网络配置,都在这个里面。
--modeldir
训练时候会写文件的目录,训练完成输出的模型保存目录
--numtrainsteps
训练多少个steps
--numeval_steps
多少个eval, 基本上两者要相差10倍以上 steps : eval
--alsologtostderr
表示日志信息
如果遇到这个错误
TypeError: can't pickle dictvalues objects
这样修改,打开model_lib.py
D:\tensorflow\models\research\objectdetection
继续训练就会很OK
启动tensorboard查看训练过程: CPU 太慢了,半天走一个step
原文发布时间为:2018-08-27
本文作者:gloomyfish
本文来自云栖社区合作伙伴“OpenCV学堂”,了解相关信息可以关注“OpenCV学堂”。
tensorflow object detection API训练公开数据集Oxford-IIIT Pets Dataset相关推荐
- tensorflow object detection API训练错误解决
问题描述 tensorflow object detection API训练coco数据集时提示错误:Windows fatal exception: access violation,如下图: Th ...
- 基于TensorFlow Object Detection API训练自己的目标识别模型
基于TensorFlow Object Detection API训练自己的目标识别模型 环境 Windows10 CUDA_9 Cudnn_9.0 Anaconda3-5.2.0 Tensorflo ...
- 使用tensorflow object detection API 训练自己的目标检测模型 (三)
在上一篇博客"使用tensorflow object detection API 训练自己的目标检测模型 (二)"中介绍了如何使用LabelImg标记数据集,生成.xml文件,经过 ...
- 关于使用tensorflow object detection API训练自己的模型-补充部分(代码,数据标注工具,训练数据,测试数据)
之前分享过关于tensorflow object detection API训练自己的模型的几篇博客,后面有人陆续碰到一些问题,问到了我解决方法.所以在这里补充点大家可能用到的东西.声明一下,本人专业 ...
- 使用tensorflow object detection API 训练自己的目标检测模型 (二)labelImg的安装配置过程
上一篇博客介绍了goggle的tensorflow object detection API 的配置和使用, 这次介绍一下如何用这个API训练一个私人定制的目标检测模型. 第一步:准备自己的数据集.比 ...
- Tensorflow object detection API训练自己的目标检测模型 详细配置教程 (一)
Tensorflow object detection API 简单介绍Tensorflow object detection API: 这个API是基于tensorflow构造的开源框架,易于构建. ...
- 使用tensorflow object detection api训练自己的数据集
简介 使用tensorflow object detection训练自己的数据集时,可能会出现 AttributeError: module 'tensorflow.contrib.data' has ...
- 如何用TF Object Detection API训练交通信号灯检测神经网络?
参加 2019 Python开发者日,请扫码咨询 ↑↑↑ 作者简介:申泽邦(Adam Shan),谷歌认证机器学习专家(Google Developer Expert),兰州大学智能驾驶团队技术负责人 ...
- 建立自己的数据集 并用Tensorflow object detection API进行训练
ps: 欢迎大家光临我的博客 建立数据集 标注工具: ubuntu 图像标注工具labelImg sudo apt-get install pyqt5-dev-tools sudo pip3 inst ...
最新文章
- anaconda在ubuntu19.10下面安装python3.6和python2.7+更换国内源+一键切换配置
- 僵尸进程的产生,危害和解决方案
- 让windows cmd也用上linux命令
- WP7中,如何把7.1的项目改为7.0的
- STC学习:光敏计数
- LoRa开发|LoRa模组AT指令开发
- 华为S9306简单实用配置合集
- selenium之滑块操作
- notepad++7.3.1中文版
- 一年经验Java面试之MySQL事务隔离级别和MVCC
- 2018年11月中旬香港, 吉隆坡8天游记
- 毕业设计 STM32单片机的智能公路收费站系统 - 物联网
- 身份证号码含义及最后一位校验算法(ISO 7064:1983.MOD 11-2)
- 孔子为何砸掉子路给工人送饭的锅?
- 【分布式】分布式环境下如何保证数据库和缓存的双写一致性?看完我明白了!!
- 基带信号、载波信号和宽带信号
- 《英语修辞与写作(修订版)》黄任(编著)epub+mobi+azw3格式下载
- 单总线CPU设计(现代时序)(HUST)
- 互联网日报 | 滴滴出租车上线“作弊举报”功能;蚂蚁集团进入上市辅导期;百度App日活达2.04亿...
- 88 java反射_4 _注解
热门文章
- docker 安装nginx_使用 Docker 在你的 mac 上搭建个服务器
- redis分布式锁实现原理_redis分布式锁实现分析与实践
- redis创建像mysql表结构_如何给redis添加新数据结构
- mysql最多多少个索引_漫画 | 一台Linux服务器最多能支撑多少个TCP连接?
- AttributeError: 'NoneType' object has no attribute 'grid'报错解决方案
- 【计算摄影】图像美学专栏上线,先从学点摄影知识开始
- 【AI初识境】被Hinton,DeepMind和斯坦福嫌弃的池化,到底是什么?​​​​​​​
- 全球及中国印刷行业发展形势与十四五应用前景分析报告2022版
- 中国城市商业银行产业模式展望及布局规模前景分析报告2021-2027年
- conn.execute的用法