caffe初探之-caffe使用Siamese Network 训练网minst数据集

来源:互联网 发布:做淘宝刷客犯法吗 编辑:程序博客网 时间:2024/05/23 18:24

MINST数据集

MNIST是一个手写数字数据库,它有60000个训练样本集和10000个测试样本集,每个样本图像的宽高为28*28。


1.获取数据集

cd caffe./data/mnist/get_mnist.sh
2.创建两个数据集

cd caffe./examples/siamese/create_mnist_siamese.sh

3.训练和测试数据

cd caffe./examples/siamese/train_mnist_siamese.sh 

注:遇到问题及解决办法

运行

create_mnist_siamese.sh
需要在caffe的安装目录下运行

需要获取root权限

需要更改caffe/examples/siamesemnist_siamese_solver.prototxt 里面的配置信息 因为电脑是cpu,所以将GPU改成CPU(要大写)


参考 http://caffe.berkeleyvision.org/gathered/examples/siamese.html

0 0
原创粉丝点击