1.得到数据集,对数据集做处理,处理成模型需要的数据格式,分出训练集,验证集以及测试集;
2.为数据集生成标签文件,生成均值文件。http://www.2cto.com/kf/201704/623027.html
3.人脸数据划分,转换为LMDB。
4.使用该数据通过以models/finetune_flickr_style/train_val.prototxt 为模板,以vgg_face_caffe/VGG_FACE_deploy.prototxt 为内容将网络结构进行填充。即加入数据输入层与改变最后一层的全连接层输出数量,修正掉旧caffe的语法。
5.拷贝models/finetune_flickr_style/solver.prototxt,并将新的针对现问题进行修改,主要修改
6.最后使用自己的数据对模型进行fine-tuning。
大部分内容参考:http://www.jianshu.com/p/a76c18a3c6d5
使用caffe微调VGG的一个例子
最新推荐文章于 2018-08-21 16:53:10 发布