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

Question

我正在参与一个项目，该项目使用了TensorFlow Models repo中的对象检测模块。我想知道只安装/管理object_detection模块的最佳方式。目前，我正在使用带有TF模型的GitHub子模块，但为了分发我的代码，我不希望捆绑整个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 .