Python 加载yolo模型不工作？

当我加载模型时，我正在构建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)

我通过以下方式构建并在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的问题在于，我应该尝试其他方法。

在python脚本中需要

将tensorflow导入为tf

。

在python脚本中需要

将tensorflow导入为tf

。

必须在Lambda层中定义

tf

试试这个：

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

---

您必须在Lambda层内定义

tf

试试这个：

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

---

您是否已安装tensorflow并在文件中导入tensorflow作为tf？请不要滥用粗体格式当然，我已安装tensorflow作为tf：Ddo您是否已安装tensorflow并在文件中导入tensorflow作为tf？当然，请不要滥用粗体格式，我已将

tensorflow

安装为

tf

：D