1.准备数据集,下面是一些图片数据下载网址:
animal:
http://www.robots.ox.ac.uk/~vgg/data/pets/
flower:
http://www.robots.ox.ac.uk/~vgg/data/flowers/
plane:
http://www.robots.ox.ac.uk/~vgg/data/airplanes_side/airplanes_side.tar
house:
http://www.robots.ox.ac.uk/~vgg/data/houses/houses.tar
guitar:
http://www.robots.ox.ac.uk/~vgg/data/guitars/guitars.tar
将这些图片数据划分为训练集和测试集
新建文件夹
我这里各选500张作为训练集,剪切到对应的文件夹。
各选300张作为测试集,剪切到对应的文件夹。
2.制作标签
比如把animal标记为第0类,flower标记为第1类,guitar第2类,house第3类,plane第4类
格式:“文件路径+标签”
制作标签程序如下:
import os
#定义caffe根目录
caffe_root='E:/graduate_student/deep