caffe学习笔记3-matcaffe接口学习
2017-03-05 10:40
190 查看
准备工作:
1.网络定义:
deploy.prototxt
train_val.prototxt
solver.prototxt
2.模型:
.caffemodel
3.数据集(train&val):
lmdb
4.均值计算:
mean
5.从caffe-master启动;
matlab
6.设置mode
CPU: caffe.set_mode_cpu(); GPU: caffe.set_mode_gpu();caffe.set_device(gpu_id);
7.设置网络结构,参数以及配置路径
net_model = ''
net_weights = ''
phase = 'test'
8.创建网络(训练)
solver = caffe.Solver('solver.prototxt路径');
solver.solve(); %开始训练
或者solver.step(1000); %只训练一千次,默认按照solver.prototxt来
iter = solver.iter(); %获得迭代次数
train_net = solver.net; %获得训练网络
test_net = solver.test_nets(1); %获得测试网络
solver.restore('.solverstate'); %防止中断,继续训练
9.创建网络(测试)
net = caffe.Net(model, weights, phase);
10.前向传播
input_data = {prepare_image(im)};
scores = net.forward(input_data);
11.后向传播
prob_diff = rand(net.blobs('prob').shape); %随机生成梯度
res = net.backward({prob_diff});
12.Reshape
net.blobs('data').reshape([227 227 3 1]);
%reshape blob 'data'
net.reshape();
13.清理nets和solvers
caffe.reset_all()
1.网络定义:
deploy.prototxt
train_val.prototxt
solver.prototxt
2.模型:
.caffemodel
3.数据集(train&val):
lmdb
4.均值计算:
mean
5.从caffe-master启动;
matlab
6.设置mode
CPU: caffe.set_mode_cpu(); GPU: caffe.set_mode_gpu();caffe.set_device(gpu_id);
7.设置网络结构,参数以及配置路径
net_model = ''
net_weights = ''
phase = 'test'
8.创建网络(训练)
solver = caffe.Solver('solver.prototxt路径');
solver.solve(); %开始训练
或者solver.step(1000); %只训练一千次,默认按照solver.prototxt来
iter = solver.iter(); %获得迭代次数
train_net = solver.net; %获得训练网络
test_net = solver.test_nets(1); %获得测试网络
solver.restore('.solverstate'); %防止中断,继续训练
9.创建网络(测试)
net = caffe.Net(model, weights, phase);
10.前向传播
input_data = {prepare_image(im)};
scores = net.forward(input_data);
11.后向传播
prob_diff = rand(net.blobs('prob').shape); %随机生成梯度
res = net.backward({prob_diff});
12.Reshape
net.blobs('data').reshape([227 227 3 1]);
%reshape blob 'data'
net.reshape();
13.清理nets和solvers
caffe.reset_all()
相关文章推荐
- 【caffe 学习笔记之2】 Win10系统下Caffe的Matlab接口设置方法与猫狗识别案例(哈士奇/松狮/波斯猫)
- 【深度学习】笔记14 windows下caffe的python接口的配置(数据的可视化环境)
- Caffe学习笔记13:数据可视化环境(python接口)配置
- 【caffe学习笔记之4】利用MATLAB接口运行cifar数据集
- caffe学习笔记4-matcaffe训练与测试
- 深度学习Caffe实战笔记(10)Windows Caffe使用MATLAB接口提取和可视化特征
- 深度学习Caffe实战笔记(12)Caffe 使用Python接口No module named google.protobuf.internal修复方案
- caffe学习笔记----4-matlab接口总结
- caffe学习笔记7-python接口绘制网络结构图
- caffe学习笔记(八)caffe+MATLAB2016接口安装
- 【caffe学习笔记之5】Win10系统下Caffe的Python接口设置方法并绘制网络结构图
- caffe学习笔记6-matlab接口总结
- [学习笔记]java Enumeration接口方法的学习
- c#学习笔记一 关于接口的几个疑问
- [学习笔记]java Enumeration接口方法的学习
- c#学习笔记之类接口
- Jpcap包的学习笔记(四)打开网络接口设备[网卡设备]
- Effective C++ 学习笔记:避免public接口出现数据成员
- Unity Application Block 学习笔记之一使用接口