两个不同的深度学习框架如何使用同一模型?

时间:2019-04-19 00:28:09

标签: python tensorflow caffe

在使用tensorflow here的深梦示例中,代码引用了Google开发的inception5h模型。但是,来自Google here的原始代码使用的是caffe而不是tensorflow,这可能是因为当时不存在张量流。同一模型可以被两个不同的框架使用的情况如何?随bvlc_googlenet.caffe模型一起分发的'deploy.prototxt'列出了许多卷积层,但是同一模型的张量流实现未引用它们,并且似乎使用了更少的层。

如果我获得了一个没有'deploy.prototxt'文件的保留模型,该如何确定模型具有多少层以及如何引用它们?

1 个答案:

答案 0 :(得分:1)

  

如果我得到的预训练模型没有'deploy.prototxt'文件,我如何确定模型有多少层

您可以使用caffe随附的draw_net.py脚本来可视化模型。