C++ (Cpp) ReadNetParamsFromBinaryFileOrDieの例

コード例 #1

ファイル: caffe_importer.cpp プロジェクト: adamrankin/opencv

    CaffeImporter(const char *pototxt, const char *caffeModel)
    {
        CV_TRACE_FUNCTION();

        ReadNetParamsFromTextFileOrDie(pototxt, &net);

        if (caffeModel && caffeModel[0])
            ReadNetParamsFromBinaryFileOrDie(caffeModel, &netBinary);
    }

コード例 #2

ファイル: solver.cpp プロジェクト: madiken/skaffe_private_old

void Solver<Dtype>::Restore(const char* state_file) {
  SolverState state;
  NetParameter net_param;
  ReadProtoFromBinaryFile(state_file, &state);
  if (state.has_learned_net()) {
    ReadNetParamsFromBinaryFileOrDie(state.learned_net().c_str(), &net_param);
    net_->CopyTrainedLayersFrom(net_param);
  }
  iter_ = state.iter();
  current_step_ = state.current_step();
  RestoreSolverState(state);
}

コード例 #3

ファイル: sgd_solver.cpp プロジェクト: fossabot/caffe

void SGDSolver<Dtype>::RestoreSolverStateFromBinaryProto(
    const string& state_file) {
  SolverState state;
  ReadProtoFromBinaryFile(state_file, &state);
  this->iter_ = state.iter();
  if (state.has_learned_net()) {
    NetParameter net_param;
    ReadNetParamsFromBinaryFileOrDie(state.learned_net().c_str(), &net_param);
    this->net_->CopyTrainedLayersFrom(net_param);
  }
  this->current_step_ = state.current_step();
  CHECK_EQ(state.history_size(), history_.size())
      << "Incorrect length of history blobs.";
  LOG(INFO) << "SGDSolver: restoring history";
  for (int i = 0; i < history_.size(); ++i) {
    history_[i]->FromProto(state.history(i));
  }
}

コード例 #4

ファイル: net.cpp プロジェクト: clarencezhang/caffe-windows-multilabels

	void Net<Dtype>::CopyTrainedLayersFrom(const string trained_filename) {
		NetParameter param;
		ReadNetParamsFromBinaryFileOrDie(trained_filename, &param);
		CopyTrainedLayersFrom(param);
	}