MergeCommand Struct Reference

Takes .prototxt and .caffemodel filenames from ARGV and fills the model with trained weights. The resulting binary model with weights to be consumed by nnc is printed to StdOut. More...

#include <MergeCommand.h>

Collaboration diagram for MergeCommand:

Public Member Functions
int	run (int argc, const char *const *argv) const override
Public Member Functions inherited from cli::Command
virtual	~Command ()=default

Detailed Description

Takes .prototxt and .caffemodel filenames from ARGV and fills the model with trained weights. The resulting binary model with weights to be consumed by nnc is printed to StdOut.

Returns

error code

Definition at line 28 of file MergeCommand.h.

Member Function Documentation

run()

int MergeCommand::run ( int  argc, const char *const *  argv  ) const

overridevirtual

Implements cli::Command.

Definition at line 29 of file MergeCommand.cpp.

{
  if (argc != 2)
  {
    std::cerr << "ERROR: this command requires exactly 2 arguments" << std::endl;
    return 254;
  }
 
  std::string model_file = argv[0];
  std::string trained_file = argv[1];
 
  // Load the network
  caffe::Net<float> caffe_test_net(model_file, caffe::TEST);
  // Load the weights
  caffe_test_net.CopyTrainedLayersFrom(trained_file);
 
  caffe::NetParameter net_param;
  caffe_test_net.ToProto(&net_param);
 
  // Write binary with initialized params into standard output
  google::protobuf::io::OstreamOutputStream os(&std::cout);
  google::protobuf::io::CodedOutputStream coded_os{&os};
 
  if (!net_param.SerializeToCodedStream(&coded_os))
  {
    std::cerr << "ERROR: Failed to serialize" << std::endl;
    return 255;
  }
  return 0;
}

Referenced by package.infer.session::inference().

---

The documentation for this struct was generated from the following files:

• compiler/caffegen/src/MergeCommand.h
• compiler/caffegen/src/MergeCommand.cpp