踩此坑的由来:
笔者写此文时欲进行 目标检测本身的训练集 进行迁移学习。
网上找到了 Tensorflow object detection API 相关的训练方法(文章应该源于2018年5月所做,有转载时间靠后的容易产生误导,可是看源码后就明了)。
笔者不幸在环境搭建上就卡了2天。
Tensorflow object detection API的github地址是:https://github.com/tensorflow/models/
笔者下载时是2019年10月。python
错误状况:
经过python下一通环境配置,
python object_detection/builders/model_builder_test.py
已经运行成功。linux
对小狗和海滩上人等的识别案例object_detection_tutorial.ipynbgit
这幅期待的图片你们看教程都很熟悉,可是就是出不来。github
迎接个人是 “服务彷佛挂掉了,可是会马上重启的.” 而后就一直卡住
从新中断服务后,再运行仍是卡在这里。ide
经过观察18年5月的object_detection_tutorial.ipynb代码和19年10月的开头就已经有较大差异。
看来只能碰碰运气回溯2018年的代码了。
从18年5月往前找有Verified标签,且开头代码部分同样的。就是这份提交了。学习
因而下载回来,放在与object_detection_tutorial.ipynb同一级目录再启动jupyter运行。
熟悉的小狗、海滩图终于出现了。优化
结语:
在Tensorflow object detection API 环境搭建上,若是这段tutorial没跑通,极可能是因为其它环境变量与linux下不一样产生的。ui
继续往下训练,如今19年10月 research/object_detection/目录下没有train.py文件了
取而代之的是model_main.py
python object_detection/model_main.py --pipeline_config_path=./data/ssdlite_mobilenet_v1_coco.config --train_dir=./train –alsologtostderr3d
可是,当前仍然有issue (https://github.com/tensorflow/models/issues/4881)
给出的建议是 使用legancy/train.py
python ./legacy/train.py --pipeline_config_path=./data/ssdlite_mobilenet_v1_coco.config --train_dir=./train –alsologtostderrblog
GPU没有启动成功,自动启动CPU,因此有点慢。自此,自备的目标检测数据集训练算是跑通了。回头继续优化实践吧,欢迎你们留言探讨。