Доступны ли весы AlexNet для Tensorflow?

Question

Тест для AlexNet является частью репозитория Tensorflow, когда я клонирую его из git. Эталон реализует слои, но мне кажется, что фактические веса для AlexNet не загружаются ни в какой точке.

Я хотел бы поиграть с Tensorflow, но мои приложения (в caffe) используют предварительно подготовленную AlexNet.

Как вы думаете, они также выпустят вес?

Answer 1

Я не знаю о существующем, но кто-то написал converter to import Caffe models into tensorflow, и вы можете найти pre-trained Alexnet models for Caffe (также см. the BVLC Model-Zoo). Я не могу обещать, что это сработает, но вы вполне можете приклеить эти два вместе, чтобы получить то, что вы хотите.

Answer 2

Нет, тест файл только реализация, и вы должны обучать и оценивать

Answer 3

http://www.cs.toronto.edu/~guerzhoy/tf_alexnet/ имеет гири как Numpy массивов обученных весов на AlexNet. Вы можете использовать их в TensorFlow.

Answer 4

Да, есть AlexNet pretrained веса, доступные для Tensorflow, вы можете скачать его here

Чтобы загрузить их в проект, вы можете использовать следующий код (адаптировано из here)

# Load the weights into memory 
weights_dict = np.load('./weight-path/bvlc_alexnet.npy', encoding='bytes').item() 

# Loop over all layer names stored in the weights dict 
for op_name in weights_dict: 

    # Check if layer should be trained from scratch 
    if op_name not in self.SKIP_LAYER: 

     with tf.variable_scope(op_name, reuse=True): 

      # Assign weights/biases to their corresponding tf variable 
      for data in weights_dict[op_name]: 

       # Biases 
       if len(data.shape) == 1: 
        var = tf.get_variable('biases', trainable=False) 
        session.run(var.assign(data)) 

       # Weights 
       else: 
        var = tf.get_variable('weights', trainable=False) 
        session.run(var.assign(data))