Caffe框架下OPenMV的机器学习(六)训练神经网络

1. 准备prototxt文件

第一章中我们谈到,我们可以借助OPenMV的源码来进一步学习OPenMV神经网络的训练方法

我们打开OPenMV源码的文件夹(没有下载的小伙伴请点击 这里

cd openmv-master\ml\cmsisnn\models\lenet

进入lenet文件夹,可以看到该文件夹下有五个文件,分别为:

lenet.network (适用于OPenMV的神经网络,是一个二进制文件)

lenet_solver.prototxt (供Caffe使用的配置训练参数的文件)

lenet_train_test.prototxt (网络各层训练和测试的参数)

test.sh (Linux脚本文件,用于测试模型)

train.sh(Linux脚本文件,用于训练模型)

 

我们把后四个文件拷贝到要网络的根目录下备用

 

2.  修改训练参数

步骤说明:

1、进入网络的根目录下

2、打开lenet_solver.prototxt文件

修改红圈的几处地方:

第一处:ne

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值