纪念第一次用 Caffe

课程设计要拿 Caffe 搞一点事情。这完全就是搞事情嘛!时间辣么短……

需求

最终需求是使用 CaffeNet,将最后的几层给修改掉,做一下 Fine Tuning,然后使用最后一层来做特征提取。

降级的需求

入门嘛…… 先能 “用起来” 再说。

加载 Mnist 数据集训练好的模型,进行一些分类,并拿出来最后一层的数据当作特征。

prototxt 文件

lenet_train_test.prototxt 进行了少许修改。

首先,最上面去掉了 mnist 层,改成了下面的形式

然后,最下面去掉 accuracyloss 层,改成一个 SoftMax 层:

其余都不动。所以,最终的 prototxt 文件是这样的:

代码

参考资料

发表评论

电子邮件地址不会被公开。 必填项已用*标注