0

我正在使用张量流的对象检测器 API 训练对象检测器。我正在按照此链接使用我自己的数据集训练对象检测器。在我开始（本地）培训过程之前，一切似乎都很好：

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

我收到以下错误：

Traceback (most recent call last):
  File "builders/model_builder_test.py", line 21, in <module>
    from object_detection.builders import model_builder
  File "/Users/vinayakpahalwan/models/research/object_detection/builders/model_builder.py", line 23, in <module>
    from object_detection.builders import matcher_builder
  File "/Users/vinayakpahalwan/models/research/object_detection/builders/matcher_builder.py", line 19, in <module>
    from object_detection.matchers import bipartite_matcher
  File "/Users/vinayakpahalwan/models/research/object_detection/matchers/bipartite_matcher.py", line 20, in <module>
    from tensorflow.contrib.image.python.ops import image_ops
ImportError: No module named 'tensorflow.contrib.image'

我现在有点卡在这里，有什么解决方法吗？

我想我已经按照此链接正确安装了对象检测 API ：

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

OK
4

1 回答 1

0

错误消息是说它找不到文件tensorflow/contrib/image。

由于您使用的是 TensorFlow 版本 0.12.1，因此查看tensorflow/contrib内部，文件夹图像尚不存在。这是从版本 1.0 引入的，如这里所示。

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

于 2017-11-09T15:21:36.670 回答