tf2 freeze冻结为pb(支持单输入和多输入)
tensorflow2.x 训练后的模型文件为 .data-00000-of-00001 和 .index,然后一般转化都是转化为pb模型和variables和assets文件,如下图所示但是可能在某种场景,需要我们使用的是freeze graph,也就是以上三个文件冻结为一个pb文件。看到有人说训练时要转成.h5文件,这个方法我没试过,但是我的模型已经训完了,再训练太麻烦了,freeze方法如下。单输入首先通过 saved_model_cli 查看图中pb文件信息根据输出信息可得,是单输入的模