本文是基于薛开宇 《学习笔记3:基于自己的数据训练和测试“caffeNet”》基础上,从头到尾把实验跑了一遍~对该文中不清楚的地方做了更正和说明。
主要工作如下:
1、下载图片建立数据库
2、将图片转化为256*256的lmdb格式
3、计算图像均值
4、定义网络修改部分参数
1、下载图片建立数据库
在caffe-master/data 下新建一个属于自己的数据库命名为babyjia,并在该文件夹下创建 train 和 val 文件夹用于存放训练集和验证集的图片:
我一共选取了100张训练集图片和20张验证集图片,分类如下:
训练集:猫 50 张图片 狗 50 张图片
验证集:猫 10 张图片 狗 10 张图片
由于训练集的图片较多,我在 train 文件夹下又增加了 dog 和 cat 文件夹
下图为训练集中狗类别的示例: