CAFFE: Fine-tuning for style recognition
2016-11-17 19:41
344 查看
假设现在有一个20类的分类问题,那么最简单的方法就是重新训练一个基于该20类的分类问题。但如果我们已经有一个1000类的模型呢?
我们其实不必大费周张的重新训练一个20类的模型,其实两个模型只有最后一层不一样,即一个的输出为1000维,另一个为20维。那么我们完全可以将1000类模型除最后一层的其余层拿过来使用,然后再使用现有数据对原模型执行fine-tuning操作,这样可以大大提高训练速度。
./build/tools/caffe train -solver models/finetune_flickr_style/solver.prototxt -weights models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel -gpu 0
如果使用cpu模式,那么最后的-gpu省略。
在套用模型的时候会将层名相同的参数值作为初始值赋值给相应层的参数,那么就要求层名相同的的层的定义要完全一致,假设输出不一致,那么赋值失败。如果在原模型中没有找到相应的层,那么该层的参数初始值与普通训练无异。
为了简单起见,作者将CAFFE_ROOT/examples/mnist/lenet_train_test.prototxt 去掉fc2层得到一个新的模型 lenet_train_test_new.prototxt
然后将对原模型进行find-tuning:
./build/tools/caffe train -solver examples/mnist/lenet_train_test_new.prototxt -weights examples/mnist/lenet.caffemodel
虽没有具体测训练时间,但训练确实是在瞬间完成的!
相关文章推荐
- 4000 caffe 教程 Fine-tuning a Pretrained Network for Style Recognition下载数据
- Caffe官方教程翻译(7):Fine-tuning for Style Recognition
- caffe 教程 Fine-tuning CaffeNet for Style Recognition on “Flickr Style” Data 遇到的问题及解决方案
- caffe学习笔记10.1--Fine-tuning a Pretrained Network for Style Recognition(new)
- Fine-tuning CaffeNet for Style Recognition on “Flickr Style” Data 问题解决
- Fine-tuning a Pretrained Network for Style Recognition
- Fine-tuning a Pretrained Network for Style Recognition
- Fine-tuning a Pretrained Network for Style Recognition
- caffe学习笔记10 -- Fine-tuning a Pretrained Network for Style Recognitio
- caffe|Fine-tuning for driver
- caffe image recognization model and prototxt for transfer learning (fine-tuning)
- Deep learning系列(十三)Transfer learning 和 caffe Fine-tuning
- Fine-tuning Convolutional Neural Networks for Biomedical Image Analysis: Actively and Incrementally
- caffe中fine-tuning的那些事
- Fine-tuning Convolutional Neural Networks for Biomedical Image Analysis: Actively and Incrementally
- caffe:fine-tuning
- Fine-tuning Convolutional Neural Networks for Biomedical Image Analysis: Actively and Incrementally
- caffe finetuning
- Fine-tuning Convolutional Neural Networks for Biomedical Image Analysis: Actively and Incrementally
- Caffe Knowledge about Softmax and fine tuning