在Mask-RCNN中在哪里创建会话?

时间:2018-08-22 05:42:06

标签: python tensorflow keras deep-learning image-segmentation

我正在学习用于对象检测和分割的深度学习。因此,我试图理解Mask-RCNN的理论以及相关代码。我遵循这个GitHub,更具体地说,遵循sample/balloon/ballon.pyhttps://github.com/matterport/Mask_RCNN

我真正不了解的一件事是创建会话的位置?我原本希望将其放在mrcnn/model.py中,但找不到。

谢谢!

1 个答案:

答案 0 :(得分:0)

Keras隐式处理会话,这意味着您基本上不需要担心。

如果要获取当前会话,可以使用:

+ time '24:00'

如果您想定义一个张量流会话并使Keras使用它来初始化所有变量:

from keras import backend as K
K.get_session().