object_detection - ImportError:无法导入名称' dataset_builder'

时间:2018-02-15 04:35:49

标签: tensorflow object-detection

我最近更新到tf 1.5并在尝试调用research / object_detection下的train.py时遇到错误说

欲了解更多信息:

Traceback (most recent call last):
  File "train.py", line 50, in <module>
    from object_detection.builders import dataset_builder
ImportError: cannot import name 'dataset_builder'

是否有任何循环依赖或者只是我得到这个错误?或者你最终一切正常吗?

2 个答案:

答案 0 :(得分:1)

如果你在代码中注意到从object_detection.builders调用了dataset_builder,这意味着你必须从一个目录(来自研究目录)执行代码。在我的例子中,我只是将train.py代码移动到研究目录,然后正常发布它就可以了!

答案 1 :(得分:1)

我有同样的问题(使用Tensorflow 1.7.0),但我通过更新models/research/object_detection文件解决了这个问题,即从repository https://github.com/tensorflow/models.git中提取最新更新。< / p>

我认为我使用的预训练模型比我的本地object_detection文件更新。