Question

我下载了vgg_19_2016_08_28.tar.gz并提取了vgg-19.pb图。我正在将其用于tf2onnx。但是，这似乎具有一些动态参数，因此如果失败，则为tf2onnx。我想检查vgg-19.pb是否为冻结图，如果不可以，如何获取冻结的vgg_19.pb图？

关于tensorflow_inception_graph的相同问题-inception_v3_2016_08_28.tar.gz

关于resnet的相同问题-resnet_v1_50_2016_08_28.tar.gz

全部从此处下载-https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models

Answer 1

要将TF模型转换为ONNX，您需要冻结图形。用于冻结图形的TensorFlow工具为https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py

例如

<configuration>
<argLine>-Dspring.profiles.active=test</argLine>
</configuration>

要查找模型开发人员将知道的TensorFlow图的输入和输出，或者您可以咨询TensorFlow的summary_graph工具（https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/graph_transforms），例如：

for i in range(len(a)):
    train[train.enrollee_id==a[i]]['enrolled_university']=train[train.enrollee_id==a[i]]['enrolled_university'].fillna(y_pred[I],inlace=True)
    print(train[train.enrollee_id==a[i]]['enrolled_university’])

vgg_19苗条模型为Frozen.pb图？

1 个答案: