Tensorflow轻量级预训练AlexNet模型

Question

7

TensorFlow slim库提供了AlexNet的图形结构，但似乎没有提供在Imagenet上预训练的AlexNet检查点(https://github.com/tensorflow/models/tree/master/research/slim)。是否有任何地方提供TF-slim预训练的AlexNet检查点？

- h1234s

我也在寻找这个... @h1234s，你有找到alexnet的checkpoint吗？ - Richard

我也是。这里有一个原版的TF链接：https://github.com/guerzh/tf_weights - txizzle

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Tin Luu · Accepted Answer

与VGG或Inception不同，TensorFlow没有预训练的AlexNet模型。Caffe有，但将其转换为TensorFlow并非易事。

感谢Frederik Kratzert，他完成了这项工作并分享了这里。

总之，您需要：

1.下载Caffe模型和权重

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

3.查看转换后的模型与slim模型之间的差异并进行适应

更多详情，请参见此处。