预训练模型集合 caffe model zoo
转自SnailTyan大佬:
https://blog.csdn.net/Quincuntial/article/details/79242342
Caffe有许多分类的预训练模型及网络结构,我自己训练过的模型总结在Github上,基本上涵盖了大部分的分类模型,包括AlexNet,VGG,GoogLeNet,Inception系列,ResNet,SENet,DenseNet,SqueezeNet。
其中会碰到不少坑,例如VGG给的结构已经太旧了,需要根据新版本Caffe的进行修改,DenseNet训练网络结构有些地方需要修改等。鉴于以上原因,我自己整理了一个Caffe
Model Zoo。
Github地址:https://github.com/SnailTyan/caffe-model-zoo
备注:大佬提供的是prototxt文件和caffemodel.将预训练模型用于微调时将.caffemodel文件作为 -weights
参数的值来使用。例如:
~/caffe $ ./build/tools/caffe train -solver solver路径/solver.protxt -weights caffemodel路径/VGG.caffemodel