python - TensorFlow slim 预训练 alexnet

Question

Tensorflow slim 库为 alexnet 提供了图形结构，但它似乎没有提供预训练的（在 Imagenet 上）alexnet 检查点（https://github.com/tensorflow/models/tree/master/research/slim） . 有没有为 TF-slim 提供预训练的 alexnet 检查点？

Answer 1

与 VGG 或 Inception 不同，TensorFlow 不附带预训练的 AlexNet。Caffe 可以，但转换为 tensorflow 并非易事。

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

总之，您需要：

1. 下载Caffe 模型和权重

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

3. 参考转换模型和苗条模型之间的差异并进行调整

更多详情，请看这里