简介
使用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