【问题标题】:where is the session created in Mask-RCNN?Mask-RCNN 中创建的会话在哪里?
【发布时间】:2019-01-28 08:06:50
【问题描述】:

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

我真的不明白的一件事是会话是在哪里创建的?我原希望在mrcnn/model.py 中有它,但我没有找到它。

谢谢!

【问题讨论】:

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


    【解决方案1】:

    Keras 隐式处理会话,这意味着您几乎不需要担心它。

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

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

    如果你想定义一个 tensorflow session 并让 Keras 用它来初始化所有变量:

    import tensorflow as tf
    sess = tf.Session()
    
    from keras import backend as K
    K.set_session(sess)
    

    【讨论】:

    • 非常感谢@sdcbr 的回答!但在你的两种情况下,我在 mask-rcnn GitHub 文件中都找不到单词 session……
    • 因为您可以使用 Keras 编写整个脚本,而无需显式处理会话。我的示例只是为了说明您可以如何做到这一点,但在大多数常见情况下并不是必需的。
    • 我的问题是我想在某处添加 :config.gpu_options.allocator_type = 'BFC' 来处理 OutOfMemory 问题……但我不知道在哪里添加,因为我看不到任何会议。此外,我仍然不明白为什么我们可以选择是否创建会话。
    • Keras 总是创建一个会话,我们不需要手动执行,甚至无法明确看到它。如果需要,可以通过K.get_session()访问当前会话
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2020-03-31
    • 2020-09-26
    • 1970-01-01
    • 2023-01-30
    相关资源
    最近更新 更多