[英]How to install or make working keras_retinanet.utils.gpu module?
I am trying to train M-RCNN using Keras and RetinaNet as in the Keras founder's example.我正在尝试使用 Keras 和 RetinaNet 训练 M-RCNN,就像在 Keras 创始人的示例中一样。 I installed retinanet package but this module isn't found.
我安装了retinanet 包,但没有找到这个模块。 May be it is related to that my TF doesn't use GPU?
可能与我的TF不使用GPU有关? How to check that?
如何检查?
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'
You probably need to install some submodules.您可能需要安装一些子模块。 The beginning of this code may help you https://www.kaggle.com/labintsevai/keras-retinanet-submission
这段代码的开头可能对你有帮助https://www.kaggle.com/labintsevai/keras-retinanet-submission
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.