在咖啡上使用VGG进行微调

Question

我正在复制http://caffe.berkeleyvision.org/gathered/examples/finetune_flickr_style.html中的步骤

我想将网络更改为VGG模型,该模型可在http://www.robots.ox.ac.uk/~vgg/software/very_deep/caffe/VGG_ILSVRC_16_layers.caffemodel获得 .

只需将模型参数替换为以下内容就足够了吗？

./build/tools/caffe train -solver models/finetune_flickr_style/solver.prototxt -weights VGG_ISLVRC_16_layers.caffemodel -gpu 0

或者我是否需要调整学习率,迭代,即它是否附带单独的原型文件？

Answer 1

您要训练的网络权重与用于初始化/微调的权重之间需要1-1对应关系.新旧模型的架构必须匹配.

VGG-16具有与models/finetune_flickr_style/train_val.prototxt(FlickrStyleCaffeNet)描述的模型不同的架构.这是解算器将尝试优化的网络.即使它没有崩溃,你加载的权重在新网络中也没有任何意义.

VGG-16网络在Caffe的Model Zoo中的此页面上的deploy.prototxt文件中进行了描述.