SqueezeNet猫狗识别

caffe训练数据准备

新建data存放训练数据,test_data存放测试数据,

data,test_data目录下新建每个类别对应的文件夹,用于存放每个类别的图片,例如,分为cat,dog两类,则分别新建目录cat,dog,

/home/data/cat
/home/data/dog

/home//test_data/cat
/home//test_data/dog

数据转换为caffe训练数据格式

训练数据,

/home//caffe/build/tools/convert_imageset –resize_height=227 –resize_width=227 –shuffle experiments/animal1/train_val/trainval/
/home/train_val/trainval/train.txt
/home/train_val/lmdb/train_lmdb45

验证数据,

/home/caffe/build/tools/convert_imageset –resize_height=227 –resize_width=227 –shuffle experiments/animal1/train_val/trainval/
/home/train_val/trainval/val.txt
/home/train_val/lmdb/val_lmdb45

下载SqueezeNet模型文件,

git clone https://github.com/DeepScale/SqueezeNet 

训练,