TensorFlow 预训练模型导入

TensorFlow 预训练模型导入

模型下载地址

Pre-trained Models和Index of /models/tf/给出了大部分较新的、性能较好的网络预训练模型，包括VGG、ResNet、各个版本的Inception等。
要测试AlexNet模型，可以到AlexNet implementation + weights in TensorFlow下载相应的.npy文件，并按照提供的.py文件重建网络，注意该网络需要特征分组训练，并且输入像素值范围为$[0,255]$。根据提供者的描述，该模型实际上由caffemodel转化而来。

【注】tensorflow是不能直接导入caffemodel的，需要将caffemodel转成tensorflow可以读取的格式，具体参见GitHub项目caffe-tensorflow。

方法

对于不同后缀名的模型，需要不同的加载方式

1. .npy文件、.npz文件、.mat文件
   这些文件保存的仅仅是网络的权重，因此为了重建网络，需要根据网络结构一步步搭建，将相应的权重赋予其中的 W,b