使用tensorflow object detection api训练自己的数据集

简介

使用tensorflow object detection训练自己的数据集时，可能会出现

AttributeError: module 'tensorflow.contrib.data' has no attribute 'parallel_interleave'  

这个是因为tensorflow(我的是1.4版本)版本的原因，没有这个函数，可以去安装tensorflow-gpu==1.8，就不会出现这个问题了。

具体安装看我的另一篇博客ubuntu下卸载cuda8.0，和安装cuda9.0，cudnn7.0,tensorflow-gpu=1.8。

训练过程

训练的过程参考这篇博客就可以了：https://blog.csdn.net/dy_guox/article/details/79111949。

还有配套的bilibili视频。也可以参考一个翻墙的一个视频教程，不过现在可以在B站看了。https://www.bilibili.com/video/av19146656