如何安装或使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.comlabintsevaikeras-retinanet-submission