7

The Tensorflow slim library provides the graph structure for alexnet, however it doesn't seem to provide a pre-trained (on Imagenet) alexnet checkpoint (https://github.com/tensorflow/models/tree/master/research/slim). Is there anywhere a pre-trained alexnet checkpoint for TF-slim is provided?

Trisoloriansunscreen
  • 1,543
  • 1
  • 15
  • 27
h1234s
  • 87
  • 1
  • 3

1 Answers1

3

Unlike VGG or Inception, TensorFlow doesn’t ship with a pretrained AlexNet. Caffe does but it's not a trivial task to convert to tensorflow.

Thanks to Frederik Kratzert, he did that job and share here

In summary, you need to:

  1. Download Caffe model and weights

  2. Use the caffe-to-tensorflow to convert the Caffe model to Tensorflow model

  3. Refer the diferences if any between the converted model and the slim model and make the adaptation

For more details, please look here

Tin Luu
  • 1,557
  • 16
  • 22