首页 > 解决方案 > 两个不同的深度学习框架如何使用相同的模型?

问题描述

此处使用tensorflow的deep dream例子中,代码参考了google开发的inception5h模型。然而,来自 google的原始代码使用的是 caffe,而不是 tensorflow,可能是因为当时不存在 tensor flow。同一个模型怎么可以被两个不同的框架使用?与 bvlc_googlenet.caffe 模型一起分发的“deploy.prototxt”列出了许多卷积层,但同一模型的张量流实现没有引用它们,而且似乎使用的层数要少得多。

如果我得到一个没有“deploy.prototxt”文件的预保留模型,我如何确定模型有多少层以及如何引用它们?

标签: pythontensorflowcaffe

解决方案


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

您可以使用draw_net.pycaffe 提供的脚本来可视化您的模型。


推荐阅读