关于caffe里面image_data_layer的使用

  • Post author:
  • Post category:其他


刚入门caffe的肯定是用demo里面的leveldb格式的数据库,说实话这数据库用起来太麻烦了,所以推荐大家入门到精通都用image_data_layer.cpp

使用方法如下所示

layer {
  name: "demo"
  type: "ImageData"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    scale: 0.00390625
    mean_value: 128
  }
  image_data_param {
    source: "data/hwdb/trainall.txt"
    root_folder: "data/hwdb/DstImg/"
    new_height: 48
    new_width: 48
    is_color: false
    batch_size: 256
    shuffle: true
  }
}

具体参数的含义,自己查看代码。其中trainall.txt的文件构造如下

每一行都是filename label,说到这里还不明白的,请结合代码自行理解,说到这份上了,应该可以了= =



版权声明:本文为kuaitoukid原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。