python - 如何安装或制作工作 keras_retinanet.utils.gpu 模块？

Question

我正在尝试使用 Keras 和 RetinaNet 训练 M-RCNN，就像 Keras 创始人的示例一样。我安装了retinanet 包，但没有找到这个模块。可能与我的 TF 不使用 GPU 有关吗？如何检查？

from keras_retinanet.utils.gpu import setup_gpu
    Traceback (most recent call last):

      File "<ipython-input-2-1f3ae263a7c6>", line 1, in <module>
        from keras_retinanet.utils.gpu import setup_gpu

    ModuleNotFoundError: No module named 'keras_retinanet.utils.gpu'

Answer 1

您可能需要安装一些子模块。此代码的开头可能会对您有所帮助https://www.kaggle.com/labintsevai/keras-retinanet-submission