如何修改从集线器恢复的 Tensorflow 2 模型的模型图和函数(不仅仅是变量值)?
(这似乎与修改恢复的 TensorFlow 模型的动机相同,但适用于 Tensorflow 2。)
特别是对于resnet_v1_101(撰写本文时的版本 4),我尝试过tensorflow_hub.KerasLayer
、tensorflow.keras.models.load_model
、tensorflow.saved_model.load
和tensorflow_hub.load
. 这些都没有导致对象具有任何功能或属性来查看流程图(更不用说修改它了)。我认为这是因为模型是从早期的 Tensorflow 移植的,或者在保存时未修饰,但是流图和函数肯定必须可用于其自己的内部计算吗?
直接与 .pb 二进制文件混在一起会让人流血。
更新:我找到了一种适合我的情况的替代方法:完全避免使用集线器并改用 tensorflow.keras.applications.ResNet101。对于未在 keras 应用程序中实现的模型,对原始问题有一个答案仍然会很好。