python - Tensorflow 对象检测 API 中的相对导入

Question

我正在尝试将tensorflow对象检测 API 用于我的个人项目之一。在这里，您可以找到指向 repo 的链接，该链接在使用之前需要克隆。在这个 repo 中，有多个目录并以这种方式排列：

models
   + research
      + object_detection
      +
    + ....
    + ....

现在object_detection是对我有用的一个。因此，我在父目录research中为我自己的模型创建了一个新目录，并尝试使用object_detection目录中提供的功能。该object_detction目录中有一个子目录，utils其中包含一些.py用于某些实用程序的文件。这是该目录中此类文件的示例链接。现在，当我尝试在我的代码中使用此文件时，它会显示no module found: object_detection. 这是一个相对导入错误，但我不知道如何解决它。我只是想确保我可以在不修改导入的情况下使用每个功能，这是可行的，因为__init__每个子目录中都有一个文件，但不知何故这不起作用。请帮忙！！

Answer 1

只是为了澄清，我假设你已经安装了 tensorflow/models。否则按照此处的说明或在虚拟环境中安装它（及其所有依赖项）。

无论哪种方式，您都可以从 object_detection API 导入代码，如下所示：

from object_detection import <myExample>

对于您的示例，这将是：

from object_detection.utils import label_map_util

Answer 2

按照安装说明，请将 slim/ 和 models/ 添加到您的 PYTHONPATH。