机器学习14:Caffe实现卷积神经网络操作流程整理

机器学习14:Caffe实现卷积神经网络操作流程整理

(1)caffe的卷积神经网络操作总流程:

         1)数据格式化处理:图片的打标和打包;

         2)编写网络结构文件:.prototxt文件,定义模型的网络结构,如层数/各层的特征图数/输入输出等。

         3)编写网络求解文件:solver.prototxt文件,设置求解网络/梯度下降参数/迭代次数等参数。网络求解文件的输入是上一步定义的网络结构。

         4)训练模型:将网络求解文件作为caffe的输入参数,调用caffe的可执行文件进行训练。

(2)文件调用过程总结:

         首先,caffe 可执行文件调用solver.prototxt网络求解文件;

         其次,网络求解文件调用网络结构文件;

         然后,网络结构文件中调用输入的训练数据图片。

         因此,训练自己的caffe模型,需要准备三个文件:数据文件(lmdb)/网络结构文件(.prototxt)/网络求解文件(solver.prototxt)。

(3)lmdb数据格式的生成:

          在caffe---example---imagenet文件夹下的一些脚本文件,如create_imagenet.sh可以帮助快速生成caffe需要的一些数据,只需要把脚本文件复制出来,稍作修改,就可以对训练图片/标注文件打包为lmdb文件。

          同时需要创建四个文件:

          1.文件夹train用于存放训练图片;

          2.文件夹val用于存放验证图片;

          3.文件train.txt用于包含每张图片的名称和对应的标签;

          4.文件val.txt用于保存图片名称和对应标签。

发布了84 篇原创文章 · 获赞 13 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_39504171/article/details/103730565
今日推荐