Tensorflow模型动物园?

时间:2015-11-15 16:53:07

标签: machine-learning neural-network tensorflow conv-neural-network

对我来说,caffe的主要优点之一是可以在自由分布的预训练模型上进行转移学习。

是否有机会从张量流格式的论文/竞赛中获得训练有素的模型?

如果没有,是否有可能将现有的caffe(或任何其他)模型转换为张量流模型?

4 个答案:

答案 0 :(得分:17)

您可以使用caffe to tensorflow model converter转换model zoo模型。如果您尝试并报告,那将是很好的知道。 converting maxpooling and padding存在潜在问题,但它似乎适用于许多模型。

答案 1 :(得分:2)

截至目前,官方TF github页面有一个list with the group of models。您也可以使用具有预训练模型的keras.applications

答案 2 :(得分:0)

使用Tensorflow进行转移学习的方法有几种,其中一些结合了TF Slim和自定义代码,但是有非常好的预训练过的Tensorflow模型Tensornets集合,其中包含几乎所有流行的模型及其预训练的权重

答案 3 :(得分:0)

有多种方法可以在TFlow中进行转移学习。 第一种解决方案是将您的保存/加载写入检查点。但是,与TF 1.0和TF 2.0不兼容

另一种是在TF之上使用一些预框架,其中一些是: TFLayers:特定于TF:TF Slim MLModels:跨平台模型动物园:MLmodels