作者:笃志单车小博_801 | 来源:互联网 | 2022-12-28 18:57
我已经在tensorflow上训练了一个pix2pix模型,并且该模型已经以检查点的形式保存了以下文件:
model-15000.meta
,model-15000.index
,model-15000.data-00000-of-00001
,graph.pbtxt
,checkpoint
。
现在,我想将其转换为protobuf文件(.pb)以进行部署。我遇到了freeze_graph.py脚本来执行此操作,但是我遇到了其中一个参数的麻烦,它是output_node_names
。
我已经尝试了几个图层名称,但是出现以下错误:
断言错误:生成器/解码器_2 / batchnorm /比例/渐变不在图中
不确定如何找到output_node_names