Keras 权重文件编辑器

[源代码]

KerasFileEditor 类

keras.saving.KerasFileEditor(filepath)

用于检查、编辑和重新保存 Keras 权重文件的实用程序。

在对模型进行架构更改后调整旧的保存的权重文件时，您会发现此类很有用。

参数

• filepath: 要检查和编辑的本地文件的路径。

示例

editor = KerasFileEditor("my_model.weights.h5")

# Displays current contents
editor.summary()

# Remove the weights of an existing layer
editor.delete_object("layers/dense_2")

# Add the weights of a new layer
editor.add_object("layers/einsum_dense", weights={"0": ..., "1": ...})

# Save the weights of the edited model
editor.resave_weights("edited_model.weights.h5")

[源代码]

summary 方法

KerasFileEditor.summary()

打印打开的文件的权重结构。

[源代码]

compare 方法

KerasFileEditor.compare(reference_model)

将打开的文件与参考模型进行比较。

此方法将列出当前打开的文件与提供的参考模型之间所有不匹配项。

参数

• reference_model: 要比较的模型实例。

返回值

• 包含以下键的字典：'status'、'error_count'、'match_count'。Status 可以是 'success' 或 'error'。'error_count' 是找到的不匹配数。'match_count' 是找到的匹配权重数。

[源代码]

save 方法

KerasFileEditor.save(filepath)

保存已编辑的权重文件。

参数

• filepath: 保存文件的路径。必须是 .weights.h5 文件。

[源代码]

rename_object 方法

KerasFileEditor.rename_object(object_name, new_name)

重命名文件中的对象（例如，层）。

参数

• object_name: 字符串，要重命名的对象的名称或路径（例如，"dense_2" 或 "layers/dense_2"）。
• new_name: 字符串，对象的新的名称。

[源代码]

delete_object 方法

KerasFileEditor.delete_object(object_name)

从文件中删除对象（例如，层）。

参数

• object_name: 字符串，要删除的对象的名称或路径（例如，"dense_2" 或 "layers/dense_2"）。

[源代码]

add_object 方法

KerasFileEditor.add_object(object_path, weights)

向文件中添加新对象（例如，层）。

参数

• object_path: 字符串，要添加的对象的完整路径（例如，"layers/dense_2"）。
• weights: 将权重名称映射到权重值（数组）的字典，例如 {"0": kernel_value, "1": bias_value}。

[源代码]

delete_weight 方法

KerasFileEditor.delete_weight(object_name, weight_name)

从现有对象中删除权重。

参数

• object_name: 字符串，要从中删除权重的对象的名称或路径（例如，"dense_2" 或 "layers/dense_2"）。
• weight_name: 字符串，要删除的权重的名称（例如，"0"）。

[源代码]

add_weights 方法

KerasFileEditor.add_weights(object_name, weights)

向现有对象添加一个或多个新权重。

参数

• object_name: 字符串，要向其添加权重的对象的名称或路径（例如，"dense_2" 或 "layers/dense_2"）。
• weights: 将权重名称映射到权重值（数组）的字典，例如 {"0": kernel_value, "1": bias_value}。