caffe网络配置--数据层

caffe网络配置--数据层

LMDB数据源

layer {

  name: "skyfer"
  type: "Data"               // Data  表示该层类型为数据层
  top/bottom: "data"      //top 表示该层的输出；bottom 表示该层的输入
  top: "label"
  include {
    phase: TRAIN/TEST   //训练模型时分训练和自测两个阶段  TRAIN 表示该层仅在训练中使用；TEST表示该层仅在自测时使用
  }
  transform_param {

    scale: 0. 0039625              //将输入的大小从0-255归一化到0-1

    mean_file:                        //可选：是否减均值

    mirror：                           //数据增强：1：开启镜像 0：不开启

    crop_size:  244                 //训练时，随机裁剪244x244大小的图像；测试时，从中间裁剪。

  }
  data_param {
    source: "F:\\caffe-master\\FER\\MyExperiment\\my_train_1x1x2048_lmdb"    //数据源
    batch_size: 64                 //越大，模型越不容易过拟合
    backend: LMDB              //数据源格式
  }

}

HDF5数据源

layer {

  name: "skyfer"
  type: " HDF5Data"               // Data  表示该层类型为数据层
  top/bottom: "data"      //top 表示该层的输出；bottom 表示该层的输入
  top: "label"
   hdf5_data_param {

    source: "F:\\caffe-master\\FER\\MyExperiment\\*.txt"    //数据源  txt文件中为hdf5数据库列表

    batch_size: 64                 //越大，模型越不容易过拟合
    backend: HDF5              //数据源格式
  }

}

IMAGE数据源

layer {

  name: "skyfer"
  type: " Image Data"               // Data  表示该层类型为数据层
  top/bottom: "data"      //top 表示该层的输出；bottom 表示该层的输入
  top: "label"
   hdf5_data_param {

    source: "F:\\caffe-master\\FER\\MyExperiment\\*.txt"    //数据源  txt文件中为image路径列表

    batch_size: 64                 //越大，模型越不容易过拟合

    new_height:

    new_width:

  }

}