当前位置: 首页>>代码示例>>C++>>正文


C++ LayerParameter::data_param方法代码示例

本文整理汇总了C++中LayerParameter::data_param方法的典型用法代码示例。如果您正苦于以下问题:C++ LayerParameter::data_param方法的具体用法?C++ LayerParameter::data_param怎么用?C++ LayerParameter::data_param使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在LayerParameter的用法示例。


在下文中一共展示了LayerParameter::data_param方法的10个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。

示例1:

AnnotatedDataLayer<Dtype>::AnnotatedDataLayer(const LayerParameter& param)
  : BasePrefetchingDataLayer<Dtype>(param),
    //reader_(param) {
    offset_() {
  db_.reset(db::GetDB(param.data_param().backend()));
  db_->Open(param.data_param().source(), db::READ);
  cursor_.reset(db_->NewCursor());
}
开发者ID:yangfly,项目名称:caffe,代码行数:8,代码来源:annotated_data_layer.cpp

示例2: QueuePair

DataReader::DataReader(const LayerParameter& param)
    : queue_pair_(new QueuePair(  //
        param.data_param().prefetch() * param.data_param().batch_size())) {
  // Get or create a body
  boost::mutex::scoped_lock lock(bodies_mutex_);
  string key = source_key(param);
  weak_ptr<Body>& weak = bodies_[key];
  body_ = weak.lock();
  if (!body_) {
    body_.reset(new Body(param));
    bodies_[key] = weak_ptr<Body>(body_);
  }
  body_->new_queue_pairs_.push(queue_pair_);
}
开发者ID:YurkovAnton,项目名称:caffe_rtpose,代码行数:14,代码来源:data_reader.cpp

示例3: return

size_t BasePrefetchingDataLayer<Ftype, Btype>::parser_threads(const LayerParameter& param) {
  // Check user's override in prototxt file
  size_t parser_threads = param.data_param().parser_threads();
  if (!auto_mode(param) && parser_threads == 0U) {
    parser_threads = 1U;  // input error fix
  }
  // 1 thread for test net
  return (auto_mode(param) || param.phase() == TEST || parser_threads == 0U) ? 1U : parser_threads;
}
开发者ID:Caffe-MPI,项目名称:Caffe-MPI.github.io,代码行数:9,代码来源:base_data_layer.cpp

示例4:

BasePrefetchingDataLayer<Dtype>::BasePrefetchingDataLayer(
    const LayerParameter& param)
    : BaseDataLayer<Dtype>(param),
      prefetch_(param.data_param().prefetch()),
      prefetch_free_(), prefetch_full_(), prefetch_current_() {
  for (int i = 0; i < prefetch_.size(); ++i) {
    prefetch_[i].reset(new Batch<Dtype>());
    prefetch_free_.push(prefetch_[i].get());
  }
}
开发者ID:huanyii,项目名称:caffe_read,代码行数:10,代码来源:base_data_layer.cpp

示例5: QueuePair

// 第一个solver里的DataReader reader_初始化的时候,
// 剩下的solvers里的DataReader reader_初始化的时候, 
// 构造函数,传入的是网络的参数、  
// 初始化queue_pair_(里面包含两个阻塞队列free_和full_)  
DataReader::DataReader(const LayerParameter& param)  
    : queue_pair_(new QueuePair(  // 1.往free_里放进去(push)了这么多个Datum. (4 x batch_size) 
        param.data_param().prefetch() * param.data_param().batch_size())) 
{  
    // Get or create a body  
    // 首先创建或者获取一个body实例  
    boost::mutex::scoped_lock lock(bodies_mutex_);  
    string key = source_key(param);// 从网络参数中获取key  
    weak_ptr<Body>& weak = bodies_[key];// bodies_是存放的string到Body的映射  
    body_ = weak.lock();  

    // 如果bodies是空的 
    if (!body_) 
    { 
        // 2.新建一个Body. 见下面分析. 
        body_.reset(new Body(param)); // 则新建Body实例到body_  
        bodies_[key] = weak_ptr<Body>(body_);// 然后存放到bodies_中去  
    }  

    // 如果只有一个source,就只有一个Body对象. Body body_的成员new_queue_pairs里的free_和full_队列中的元素为Datum, 队列长度为 (4 x batch_size) 
    body_->new_queue_pairs_.push(queue_pair_); // 并将queue_pair放入body_中的new_queue_pairs_中去  
}  
开发者ID:runaway,项目名称:studycaffe,代码行数:26,代码来源:data_reader.cpp

示例6:

 // A source is uniquely identified by its layer name + path, in case
 // the same database is read from two different locations in the net.
 static inline string source_key(const LayerParameter& param, bool is_nv_data) {
   return param.name() + ":" +
     (is_nv_data ? param.nvdata_param().source() : param.data_param().source());
 }
开发者ID:XinLiuNvidia,项目名称:caffe,代码行数:6,代码来源:data_reader.hpp

示例7:

 // A source is uniquely identified by its layer name + path, in case
 // the same database is read from two different locations in the net.
 static inline string source_key(const LayerParameter& param) {
   return param.name() + ":" + param.data_param().source();
 }
开发者ID:1064no1carry,项目名称:caffe,代码行数:5,代码来源:data_reader.hpp

示例8: QueuePair

DataReader::DataReader(const LayerParameter& param) :param(param),cur_idx(0){
	pair.reset(new QueuePair(param.data_param().prefetch()*param.data_param().batch_size()));
	parseImageset();
	startThread();
}
开发者ID:dongxf369,项目名称:Dragon,代码行数:5,代码来源:data_reader.cpp

示例9:

bool BasePrefetchingDataLayer<Ftype, Btype>::auto_mode(const LayerParameter& param) {
  // Both should be set to positive for manual mode
  const DataParameter& dparam = param.data_param();
  bool auto_mode = !dparam.has_threads() && !dparam.has_parser_threads();
  return auto_mode;
}
开发者ID:Caffe-MPI,项目名称:Caffe-MPI.github.io,代码行数:6,代码来源:base_data_layer.cpp

示例10: DataSiameseLayer

 explicit DataSiameseLayer(const LayerParameter& param): BaseDataLayer<Dtype>(param), data_param_(param.data_param()) {}
开发者ID:yuxng,项目名称:caffe,代码行数:1,代码来源:data_siamese_layer.hpp


注:本文中的LayerParameter::data_param方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。