caffe prototxt 中的多种数据输入方式

1,MLDB模式

name: "CaffeNet"
layer {
  name: "data"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    mirror: true
    crop_size: 227
    mean_file: "data/ilsvrc12/imagenet_mean.binaryproto"
  }
# mean pixel / channel-wise mean instead of mean image
#  transform_param {
#    crop_size: 227
#    mean_value: 104
#    mean_value: 117
#    mean_value: 123
#    mirror: true
#  }
  data_param {
    source: "examples/imagenet/ilsvrc12_train_lmdb"
    batch_size: 256
    backend: LMDB
  }
}

2,图片模式

layer {
  name: "data"
  type: "ImageData"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  transform_param {
    mirror: false
    crop_size: 227
    mean_file: "data/ilsvrc12/imagenet_mean.binaryproto"
  }
  image_data_param {
    source: "data/ilsvrc12/val.txt"
    root_folder:"/home/lili/ILSVRC2012_img_val/"
    new_height:256
    new_width:256
    is_color:true
    batch_size: 50
    shuffle:false
  }
}

其中val.txt的格式为:
这里写图片描述
一般在caffe的data/ilsvrc12/的目录下

阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

caffe prototxt 中的多种数据输入方式

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭