使用Tensorflow的对象检测API使用我自己的数据集训练对象检测器时出错

Question

我正在用tensorflow的对象检测器API训练一个对象检测器。我跟随this link训练一个拥有我自己数据集的对象检测器。在我开始培训之前，一切似乎都很好（在当地）：

python /Users/vinayakpahalwan/models/research/object_detection/train.py --logtostderr --train_dir=/Users/vinayakpahalwan/Desktop/ObjectDetection/training/ --pipeline_config_path=/Users/vinayakpahalwan/Desktop/ObjectDetection/training/ssd_mobilenet_v1_coco.config

我得到以下错误：

^{pr2}$

我现在有点困在这里了，有什么解决办法吗？在

我想我已经按照this link正确安装了对象检测API：

Vinayaks-MacBook-Air:object_detection vinayakpahalwan$ python builders/model_builder_test.py
...........
----------------------------------------------------------------------
Ran 11 tests in 0.070s

OK

Answer 1

错误消息表示无法找到文件tensorflow/contrib/image。在

由于您使用的是Tensorflow版本0.12.1，查看tensorflow/contrib，文件夹image尚不存在。这是从1.0版引入的，如here。在

此时您的解决方法是更新TS