如何使用leveldb,我可以在pycaffe接口中使用什么样的数据层?

2024-06-01 06:39:10 发布

您现在位置:Python中文网/ 问答频道 /正文

我试着做火车/val.prototxt文件使用leveldb by caffe python接口:

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接口中,我找不到合适的数据层python包装器(例如L.MemoryData),尽管我试图在BLVC/caffe页面中找到示例和教程。你知道吗

你能注意到我可以用哪一层吗?你知道吗


Tags: 文件homedatasizeparamtopvalcaffe
1条回答
网友
1楼 · 发布于 2024-06-01 06:39:10

使用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>的层。你知道吗

相关问题 更多 >