如何安装或制作可工作的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