tensorflow - MaskRCNN 转成 pb 文件

Question

我已经训练了一个 MaskrCNN 模型，并尝试通过以下代码将其转换为 pb 文件：

    import os
    import sys
    import warnings

    import keras.backend as K
    import tensorflow.compat.v1 as tf
    tf.disable_v2_behavior()
    tf.reset_default_graph() 
    tf.compat.v1.reset_default_graph()
    warnings.filterwarnings('ignore', category=FutureWarning)
    os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
    ROOT_DIR = os.getcwd()
    sys.path.append(ROOT_DIR)  # To find local version of the library
    from mrcnn import model as modellib
    from mrcnn import utils
    from tensorflow.python.framework import ops
    ops.reset_default_graph()
    tf.reset_default_graph() 
    session = tf.InteractiveSession()
    K.clear_session()
    K.set_learning_phase(0)

运行它时，我收到以下错误：

    AttributeError: module 'tensorflow' has no attribute 'reset_default_graph'

我的 tensorflow 版本是 2.2.0，我的 keras 版本是 2.2.4 我有点困惑，因为根据版本有很多提示和注释。我该如何克服这个错误？我的下一个问题是如何使用 pb 测试图像？

标签： tensorflow