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


C++ NetParameter::add_layers方法代码示例

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


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

示例1: main

// Generate inception network specifications.
int main(int argc, char** argv) {
  string prefix(argv[1]);

  // Deployment network spec.
  string deploy_fname = prefix + "_deploy.prototxt";
  NetParameter deploy;
  deploy.set_name("Inception");
  deploy.add_input("data");
  deploy.add_input_dim(10);
  deploy.add_input_dim(3);
  deploy.add_input_dim(227);
  deploy.add_input_dim(227);
  AddBody(&deploy);
  AddSoftmaxLayer("prob", "linear", deploy.add_layers());
  WriteProtoToTextFile(deploy, deploy_fname);

  // Training network spec.
  string train_fname = prefix + "_train.prototxt";
  NetParameter train;
  train.set_name("Inception");
  AddDataLayer("data", "label", traindb, meanfile, 64, 227, true,
               train.add_layers());
  AddBody(&train);
  AddSoftmaxLoss("prob", "linear", "label", train.add_layers());
  WriteProtoToTextFile(train, train_fname);

  // Validation network spec.
  string val_fname = prefix + "_val.prototxt";
  NetParameter val;
  val.set_name("Inception");
  AddDataLayer("data", "label", valdb, meanfile, 50, 227, false,
               val.add_layers());
  AddBody(&val);
  AddSoftmaxLayer("prob", "linear", val.add_layers());
  AddAccuracyLayer("accuracy", "prob", "label", val.add_layers());
  WriteProtoToTextFile(val, val_fname);
}
开发者ID:JaneShenYY,项目名称:fgs-obj,代码行数:38,代码来源:gen_inception.cpp

示例2: to_proto

 void BaseProducer::to_proto (NetParameter& net) const {
   *net.add_layers() = layer_param_;
 }
开发者ID:SiyuZhu,项目名称:CircuitRecognition,代码行数:3,代码来源:producers.cpp


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