【深度学习框架Caffe学习与应用】第二课

时间:2022-12-03 20:13:44

1.使用Protocol Buffer提供的命令行编译工具protoc进行编译

首先写好了一个proto文件:


【深度学习框架Caffe学习与应用】第二课


【深度学习框架Caffe学习与应用】第二课


然后将该文件编译成一个C++的输出


【深度学习框架Caffe学习与应用】第二课


此时可以看到,test文件夹中出现了.cc跟.h文件,然后使用test_pb.cc测试生成的头文件ly.helloworld.proto


【深度学习框架Caffe学习与应用】第二课


然后去编译这个文件


【深度学习框架Caffe学习与应用】第二课


这样就操作成功了,此时运行./test_pb,就能输出了


附:我在根目录下新建了一个test文件夹,里面放了写好的几个文件,目录如下:


【深度学习框架Caffe学习与应用】第二课


2.训练mnist数据集


第一步:获取数据

首先进入caffe根目录


【深度学习框架Caffe学习与应用】第二课


可以看到,下载了四个文件


第二步:将数据转化为lmdb格式


【深度学习框架Caffe学习与应用】第二课


文件夹中就生成了下图所示:


【深度学习框架Caffe学习与应用】第二课


第三步:训练


【深度学习框架Caffe学习与应用】第二课


【深度学习框架Caffe学习与应用】第二课

【深度学习框架Caffe学习与应用】第二课

训练完毕