zoukankan      html  css  js  c++  java
  • caffe的python接口学习(1)生成配置文件

    ---恢复内容开始---

    看了denny的博客,写下自己觉得简短有用的部分

    想用caffe训练数据首先要学会编写配置文件:

    (即便是用别人训练好的模型也要进行微调的,所以此关不可跨越)

    代码就不粘贴了,我只总结文字部分的内容:

    1 写好各文件/数据的路径;

    2 创建数据层,向上传递图像数据和标签数据;

    3 卷积层;(创建层主要就是输入参数,要不怎么体现框架带来的好处,哈哈)

    4 创建激活函数;

    5 创建池化层;

    6 创建一个全链接层;(将各单元学到的经验结合到一起,降维求特征和)

    7 创建一个dropout层;(防止过拟合,强迫一个神经元和随机挑选出来的其他神经元共同工作,减弱神经元节点间的联合适应性,增强了泛化能力)

    8 创建一个softmax层;(计算与标注样本的差距)

    9 训练时不需要accuracy层,但验证阶段是需要的;(预测结果与输入标签的对比)

    10 将以上设置写入配置文件prototxt文件;

    通过上面这个文件的执行,我们就会得到两个配置文件:train.prototxt和val.prototxt,分别用于训练阶段和验证阶段。

    这种方式生成配置文件,必须有个前提,就是要先把原始图片转换成LMDB文件才行。如果我们已经把原始图片做成了一个列表清单(txt文件,一行一张图片),则可以不用LMDB格式作为输入数据,可以用ImageData作为数据源输入,即第一层由原来的Data类型,变成了ImageData类型,不需要LMDB文件和均值文件,但需要一个txt文件。

    ---恢复内容结束---

  • 相关阅读:
    Spring入门之一-------实现一个简单的IoC
    SpringBoot+SpringSecurity之多模块用户认证授权同步
    SpringBoot+SpringSecurity之如何forword到登录页面
    SpringBoot实现OAuth2认证服务器
    SpringBoot安全认证Security
    SpringBoot的ApplicationRunner和CommandLineRunner
    SpringBoot通过ApplicationArguments获取args
    安利demo
    多路canvas的mapbox gl
    复合canvas叠加显示
  • 原文地址:https://www.cnblogs.com/niulang/p/8984189.html
Copyright © 2011-2022 走看看