Darknet/YOLO v6.0-37-gb57f9029
Object Detection Framework
 
Loading...
Searching...
No Matches
Darknet::Node Class Referencefinal

Class used to create and manipulate ONNX nodes. More...

#include "darknet_node.hpp"

Collaboration diagram for Darknet::Node:

Public Member Functions

 Node ()=delete
 
 Node (const Darknet::CfgSection &section, const Darknet::VInt &v)
 Constructor for many int constants.
 
 Node (const Darknet::CfgSection &section, const float f, const size_t bit_size=32)
 Constructor for a single float constant.
 
 Node (const Darknet::CfgSection &section, const std::string &append="")
 Constructor.
 
 Node (const Node &rhs)=delete
 
 Node (const std::string &n)
 Constructor.
 
 ~Node ()
 Destructor.
 
Nodeadd_attribute_FLOAT (const std::string &key, const float val)
 
Nodeadd_attribute_INT (const std::string &key, const int val)
 
Nodeadd_attribute_INTS (const std::string &key, const Darknet::VInt &val)
 
Nodeadd_attribute_STR (const std::string &key, const std::string &val)
 
Nodeadd_input (const std::string &input)
 
Nodeadd_input (int idx)
 Look up the node's name from the given index and use that as input.
 
Nodedoc (const Darknet::CfgSection &section)
 
Nodedoc (const std::string &str)
 
Nodedoc_append (const std::string &str)
 
Nodeinit (const std::string &n)
 
Nodeoperator= (const Node &rhs)=delete
 
Nodeset_output (const std::string &out="")
 
Nodetype (const std::string &type)
 

Static Public Member Functions

static std::string get_output_for_layer_index (const int idx)
 

Public Attributes

size_t counter
 
size_t layer_index
 
std::string name
 
onnx::NodeProtonode
 
std::string output
 

Static Public Attributes

static std::string cfg_filename = "unknown"
 
static onnx::GraphProtograph = nullptr
 
static size_t node_counter = 0
 
static Darknet::MIdStr output_per_layer_index
 Keep track of the most recent output name for each of the layers.
 

Detailed Description

Class used to create and manipulate ONNX nodes.

Constructor & Destructor Documentation

◆ Node() [1/6]

Darknet::Node::Node ( )
delete

◆ Node() [2/6]

Darknet::Node::Node ( const Node rhs)
delete

◆ Node() [3/6]

Darknet::Node::Node ( const std::string &  n)

Constructor.

Here is the call graph for this function:

◆ Node() [4/6]

Darknet::Node::Node ( const Darknet::CfgSection section,
const std::string &  append = "" 
)

Constructor.

Here is the call graph for this function:

◆ Node() [5/6]

Darknet::Node::Node ( const Darknet::CfgSection section,
const float  f,
const size_t  bit_size = 32 
)

Constructor for a single float constant.

Here is the call graph for this function:

◆ Node() [6/6]

Darknet::Node::Node ( const Darknet::CfgSection section,
const Darknet::VInt v 
)

Constructor for many int constants.

Here is the call graph for this function:

◆ ~Node()

Darknet::Node::~Node ( )

Destructor.

Member Function Documentation

◆ add_attribute_FLOAT()

Darknet::Node & Darknet::Node::add_attribute_FLOAT ( const std::string &  key,
const float  val 
)
Here is the call graph for this function:

◆ add_attribute_INT()

Darknet::Node & Darknet::Node::add_attribute_INT ( const std::string &  key,
const int  val 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ add_attribute_INTS()

Darknet::Node & Darknet::Node::add_attribute_INTS ( const std::string &  key,
const Darknet::VInt val 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ add_attribute_STR()

Darknet::Node & Darknet::Node::add_attribute_STR ( const std::string &  key,
const std::string &  val 
)
Here is the call graph for this function:

◆ add_input() [1/2]

Darknet::Node & Darknet::Node::add_input ( const std::string &  input)

◆ add_input() [2/2]

Darknet::Node & Darknet::Node::add_input ( int  idx)

Look up the node's name from the given index and use that as input.

Here is the caller graph for this function:

◆ doc() [1/2]

Darknet::Node & Darknet::Node::doc ( const Darknet::CfgSection section)

◆ doc() [2/2]

Darknet::Node & Darknet::Node::doc ( const std::string &  str)
Here is the caller graph for this function:

◆ doc_append()

Darknet::Node & Darknet::Node::doc_append ( const std::string &  str)

◆ get_output_for_layer_index()

std::string Darknet::Node::get_output_for_layer_index ( const int  idx)
static
Here is the caller graph for this function:

◆ init()

Darknet::Node & Darknet::Node::init ( const std::string &  n)
Here is the caller graph for this function:

◆ operator=()

Node & Darknet::Node::operator= ( const Node rhs)
delete

◆ set_output()

Darknet::Node & Darknet::Node::set_output ( const std::string &  out = "")

◆ type()

Darknet::Node & Darknet::Node::type ( const std::string &  type)
Here is the caller graph for this function:

Member Data Documentation

◆ cfg_filename

std::string Darknet::Node::cfg_filename = "unknown"
static

◆ counter

size_t Darknet::Node::counter

◆ graph

onnx::GraphProto * Darknet::Node::graph = nullptr
static

◆ layer_index

size_t Darknet::Node::layer_index

◆ name

std::string Darknet::Node::name

◆ node

onnx::NodeProto* Darknet::Node::node

◆ node_counter

size_t Darknet::Node::node_counter = 0
static

◆ output

std::string Darknet::Node::output

◆ output_per_layer_index

Darknet::MIdStr Darknet::Node::output_per_layer_index
static

Keep track of the most recent output name for each of the layers.


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