Data

class dragon.vm.caffe.core.layers.Data(layer_param)[source]

Load batch of data for image classification.

Examples:

layer {
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  data_param {
    source: "/data/train"
    batch_size: 128
    prefetch: 4
  }
  image_data_param {
    shuffle: true
  }
  transform_param {
    mirror: true
    crop_size: 224
    mean_value: 104.00698793
    mean_value: 116.66876762
    mean_value: 122.67891434
  }
}
layer {
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  data_param {
    source: "/data/val"
    batch_size: 64
  }
  transform_param {
    crop_size: 224
    mean_value: 104.00698793
    mean_value: 116.66876762
    mean_value: 122.67891434
  }
}