Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Tensorflow slim pre-trained alexnet [closed]

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?

like image 847
h1234s Avatar asked Jan 30 '17 17:01

h1234s


1 Answers

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

like image 99
Tin Luu Avatar answered Oct 05 '22 20:10

Tin Luu