python - 如何使用leveldb以及我可以在pycaffe界面中使用什么样的dataLayer？

Question

我尝试使用 leveldb by caffe python 接口(interface)制作 train/val.prototxt:

layer {
  name: "cifar"
  type: "Data"
  top: "data"
  top: "label"
  data_param {
    source: "/home/youngwan/data/cifar10/cifar10-gcn-leveldb-splits/cifar10_full_train_leveldb_padded"
    batch_size: 100
    backend: LEVELDB
  }
  transform_param {
    mean_file: "/home/youngwan/data/cifar10/cifar10-gcn-leveldb-splits/paddedmean.binaryproto"
    mirror: 1
    crop_size: 32
  }
  include: { phase: TRAIN }
}

但是在caffe python界面中，即使我尝试在BLVC/caffe页面中查找示例和教程，我也找不到合适的数据层python包装器(例如L.MemoryData)。

你能注意到我可以使用哪个'L.xxx'层吗？

Answer 1

使用caffe.NetSpec()界面，你可以拥有你想要的所有层:

from caffe import layers as L, params as P
cifar = L.Data(data_param={'source': '/home/youngwan/data/cifar10/cifar10-gcn-leveldb-splits/cifar10_full_train_leveldb_padded', 
                           'batch_size': 100,
                           'backend': P.Data.LEVELDB},
               transform_param={'mean_file': '/home/youngwan/data/cifar10/cifar10-gcn-leveldb-splits/paddedmean.binaryproto',
                                'mirror': 1,
                                'crop_size': 32},
               include={'phase':caffe.TRAIN})

基本上，L.<layer type>定义类型 <layer type> 的层.