正在加载yolo模型无法正常工作？

Question

加载模型时，我正在构建CNN yolo模型

json_file = open('model.json', 'r')

loaded_model_json = json_file.read()

loaded_model = model_from_json(loaded_model_json)

我遇到一个奇怪的错误：

NameError: name 'tf' is not defined

然后我意识到它是从某种方法中产生的

def space_to_depth_x2(x):
    return tf.space_to_depth(x, block_size=2)

我通过以下方法构建并在The cnn模型中使用它：

skip_connection = BatchNormalization(name='norm_21')(skip_connection)
skip_connection = LeakyReLU(alpha=0.1)(skip_connection)
skip_connection = Lambda(space_to_depth_x2)(skip_connection)

所以我认为Lambda有问题，应该尝试其他方法吗？

Answer 1

您必须在Lambda层中定义tf！

尝试一下：

def space_to_depth_x2(x):
    import tensorflow as tf
    return tf.space_to_depth(x, block_size=2)

Answer 2

您的Python脚本中需要import tensorflow as tf。