生成PASCAL TFRecord文件时的tensorflow对象检测API ImportError答案

【问题标题】：tensorflow object detection API ImportError when generating PASCAL TFRecord files生成PASCAL TFRecord文件时的tensorflow对象检测API ImportError
【发布时间】：2017-07-22 13:13:23
【问题描述】：

我正在尝试使用 Tensorflow 对象检测 API，并且我已经成功测试了安装，但我们尝试使用给定命令生成 PASCAL VOC TFRecord 文件

python object_detection/create_pascal_tf_record.py \
--label_map_path=object_detection/data/pascal_label_map.pbtxt \
--data_dir=VOCdevkit --year=VOC2012 --set=train \
--output_path=pascal_train.record

我遇到了以下错误：

Traceback (most recent call last):
  File "object_detection/create_pascal_tf_record.py", line 36, in <module>
    from object_detection.utils import dataset_util
ImportError: No module named object_detection.utils

我的 PYTHONPATH 是：

:/usr/local/lib/python2.7/dist-packages/tensorflow/models:/usr/local/lib/python2.7/dist-packages/tensorflow/models/slim

我在 /models 目录中运行上述命令，有谁知道如何解决这个问题？

【问题讨论】：

我在 python 控制台中运行'from object_detection.utils import dataset_util'，它工作正常。我真的很困惑！

标签： tensorflow importerror object-detection

【解决方案1】：

我遇到了同样的问题，我通过添加解决了它：

import os
import sys
sys.path.append(os.path.abspath("./object_detection"))

和

from object_detection.utils import dataset_util

变成

from utils import dataset_util

【讨论】：

【解决方案2】：

最好将 'object_detection' 添加到 env 路径中，而不是在 python 脚本中。

您可以在 shell 中的“object_detection”目录中执行此操作。

export PYTHONPATH=$PYTHONPATH:`pwd`

或者添加.bashrc/.zshrc的路径（取决于你的shell），所以不需要每次都导出。

echo "export PYTHONPATH=$PYTHONPATH:`pwd` >> ~/.bashrc

【讨论】：