分发python包时如何管理object_detection包依赖？

Question

我正在一个使用 TensorFlow 模型存储库中的对象检测模块的项目中工作。 我想知道仅安装/管理 object_detection 模块的最佳方法。 目前，我将 github 子模块与 TF 模型一起使用，但是为了分发我的代码，我不希望捆绑整个 TF 模型存储库，只捆绑 object_detection 模块。

有可能这样做吗？

提前致谢。

Answer 1

我建议的方法是继续使用该项目作为 git 子模块，因为object_detection模块取决于research目录之外的official包。 此外，目前 pip 上没有更新版本的object_detection 。

如果你真的只想使用你需要的部分，你可以按照@Alexandre Passos 提到的方法进行操作。

您可以添加一个 Makefile 来帮助您分发包，并添加一个setup.py来指定其他依赖项并使您的包可分发。

Makefile 应如下所示：

init:
    git submodule update --init --progress
    cd models/research && protoc object_detection/protos/*.proto --python_out=.
    cd models/research && python setup.py install 
    cd models/research/slim && python setup.py install 
    # Install your package
    pip install -e .