AlexNet权重是否适用于Tensorflow?

时间:2015-11-12 14:14:41

标签: deep-learning tensorflow

当我从git克隆它时,AlexNet的基准测试是Tensorflow存储库的一部分。基准测试实现了图层,但我认为AlexNet的实际权重在任何时候都没有加载。

我想玩Tensorflow,但我的应用程序(在caffe中)正在使用预先训练的AlexNet。

你认为他们也会释放重量吗?

4 个答案:

答案 0 :(得分:6)

我不知道现有的,但有人写了converter to import Caffe models into tensorflow,你可以找到pre-trained Alexnet models for Caffe(另见the BVLC Model-Zoo)。我不能保证它会起作用,但你很可能将这两者粘在一起以获得你想要的东西。

答案 1 :(得分:1)

http://www.cs.toronto.edu/~guerzhoy/tf_alexnet/在AlexNet上的权重是训练重量的numpy数组。您可以在TensorFlow中使用它们。

答案 2 :(得分:1)

是的,有可用于Tensorflow的AlexNet预训练重量,您可以下载它here

要将它们加载到项目中,您可以使用以下代码(改编自here

class ApiNameController < ApplicationController
  skip_before_action :authenticate_user!, only: :get_request

答案 3 :(得分:0)

不,基准文件只是实现,你必须训练和评估