如何安装或制作工作keras_retinanet.utils.gpu模块?

Question

我正在尝试使用Keras和RetinaNet训练M-RCNN，就像Keras创始人的例子一样。我安装了视网膜包，但找不到此模块。可能与我的 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