Tensorflow苗条预训练的亚历山大

时间:2017-01-30 17:05:00

标签: python tensorflow tf-slim

Tensorflow slim库提供了alexnet的图形结构,但它似乎没有提供预训练的(在Imagenet上)alexnet检查点(https://github.com/tensorflow/models/tree/master/research/slim)。有没有提供预先训练的TF-slim的alexnet检查点?

1 个答案:

答案 0 :(得分:2)

与VGG或Inception不同,TensorFlow不附带预先训练的AlexNet。 Caffe确实如此,但转换为tensorflow并不是一项微不足道的任务。

感谢Frederik Kratzert,他完成了这项工作并分享了here

总之,您需要:

  1. 下载Caffe model and weights

  2. 使用caffe-to-tensorflow将Caffe模型转换为Tensorflow模型

  3. 请参阅转换模型与纤薄模型之间的差异(如果有)并进行调整

  4. 有关详细信息,请查看here