OpenCV
3.2.0
Open Source Computer Vision
|
This class allows to create and manipulate comprehensive artificial neural networks. More...
#include "dnn.hpp"
Public Types | |
typedef DictValue | LayerId |
Container for strings and integers. More... | |
Public Member Functions | |
Net () | |
Default constructor. More... | |
~Net () | |
Destructor frees the net only if there aren't references to the net anymore. More... | |
int | addLayer (const String &name, const String &type, LayerParams ¶ms) |
Adds new layer to the net. More... | |
int | addLayerToPrev (const String &name, const String &type, LayerParams ¶ms) |
Adds new layer and connects its first input to the first output of previously added layer. More... | |
void | allocate () |
Initializes and allocates all layers. More... | |
void | connect (String outPin, String inpPin) |
Connects output of the first layer to input of the second layer. More... | |
void | connect (int outLayerId, int outNum, int inpLayerId, int inpNum) |
Connects #outNum output of the first layer to #inNum input of the second layer. More... | |
void | deleteLayer (LayerId layer) |
Delete layer for the network (not implemented yet) More... | |
bool | empty () const |
void | forward (LayerId toLayer=String()) |
Runs forward pass to compute output of layer toLayer . More... | |
void | forward (LayerId startLayer, LayerId toLayer) |
Runs forward pass to compute output of layer toLayer , but computations start from startLayer . More... | |
void | forward (const std::vector< LayerId > &startLayers, const std::vector< LayerId > &toLayers) |
void | forwardOpt (LayerId toLayer) |
Optimized forward. More... | |
void | forwardOpt (const std::vector< LayerId > &toLayers) |
Blob | getBlob (String outputName) |
Returns the layer output blob. More... | |
Ptr< Layer > | getLayer (LayerId layerId) |
Returns pointer to layer with specified name which the network use. More... | |
int | getLayerId (const String &layer) |
Converts string name of the layer to the integer identifier. More... | |
std::vector< String > | getLayerNames () const |
Blob | getParam (LayerId layer, int numParam=0) |
Returns parameter blob of the layer. More... | |
void | setBlob (String outputName, const Blob &blob) |
Sets the new value for the layer output blob. More... | |
void | setNetInputs (const std::vector< String > &inputBlobNames) |
Sets outputs names of the network input pseudo layer. More... | |
void | setParam (LayerId layer, int numParam, const Blob &blob) |
Sets the new value for the learned param of the layer. More... | |
This class allows to create and manipulate comprehensive artificial neural networks.
Neural network is presented as directed acyclic graph (DAG), where vertices are Layer instances, and edges specify relationships between layers inputs and outputs.
Each network layer has unique integer id and unique string name inside its network. LayerId can store either layer name or layer id.
This class supports reference counting of its instances, i. e. copies point to the same instance.
typedef DictValue cv::dnn::Net::LayerId |
Container for strings and integers.
cv::dnn::Net::Net | ( | ) |
Default constructor.
cv::dnn::Net::~Net | ( | ) |
Destructor frees the net only if there aren't references to the net anymore.
int cv::dnn::Net::addLayer | ( | const String & | name, |
const String & | type, | ||
LayerParams & | params | ||
) |
Adds new layer to the net.
name | unique name of the adding layer. |
type | typename of the adding layer (type must be registered in LayerRegister). |
params | parameters which will be used to initialize the creating layer. |
int cv::dnn::Net::addLayerToPrev | ( | const String & | name, |
const String & | type, | ||
LayerParams & | params | ||
) |
Adds new layer and connects its first input to the first output of previously added layer.
void cv::dnn::Net::allocate | ( | ) |
Initializes and allocates all layers.
Connects output of the first layer to input of the second layer.
outPin | descriptor of the first layer output. |
inpPin | descriptor of the second layer input. |
Descriptors have the following template <layer_name>[.input_number]
:
layer_name
is sting name of the added layer. If this part is empty then the network input pseudo layer will be used;the second optional part of the template input_number
is either number of the layer input, either label one. If this part is omitted then the first layer input will be used.
void cv::dnn::Net::connect | ( | int | outLayerId, |
int | outNum, | ||
int | inpLayerId, | ||
int | inpNum | ||
) |
Connects #outNum
output of the first layer to #inNum
input of the second layer.
outLayerId | identifier of the first layer |
inpLayerId | identifier of the second layer |
outNum | number of the first layer output |
inpNum | number of the second layer input |
void cv::dnn::Net::deleteLayer | ( | LayerId | layer | ) |
Delete layer for the network (not implemented yet)
bool cv::dnn::Net::empty | ( | ) | const |
Returns true if there are no layers in the network.
Runs forward pass to compute output of layer toLayer
.
By default runs forward pass for the whole network.
Runs forward pass to compute output of layer toLayer
, but computations start from startLayer
.
void cv::dnn::Net::forward | ( | const std::vector< LayerId > & | startLayers, |
const std::vector< LayerId > & | toLayers | ||
) |
This is an overloaded member function, provided for convenience. It differs from the above function only in what argument(s) it accepts.
void cv::dnn::Net::forwardOpt | ( | LayerId | toLayer | ) |
Optimized forward.
Makes forward only those layers which weren't changed after previous forward().
void cv::dnn::Net::forwardOpt | ( | const std::vector< LayerId > & | toLayers | ) |
This is an overloaded member function, provided for convenience. It differs from the above function only in what argument(s) it accepts.
Returns the layer output blob.
outputName | the descriptor of the returning layer output blob. |
Returns pointer to layer with specified name which the network use.
int cv::dnn::Net::getLayerId | ( | const String & | layer | ) |
Converts string name of the layer to the integer identifier.
std::vector<String> cv::dnn::Net::getLayerNames | ( | ) | const |
Returns parameter blob of the layer.
layer | name or id of the layer. |
numParam | index of the layer parameter in the Layer::blobs array. |
Sets the new value for the layer output blob.
outputName | descriptor of the updating layer output blob. |
blob | new blob. |
blob
must have the same shape, because network reshaping is not implemented yet. void cv::dnn::Net::setNetInputs | ( | const std::vector< String > & | inputBlobNames | ) |
Sets outputs names of the network input pseudo layer.
Each net always has special own the network input pseudo layer with id=0. This layer stores the user blobs only and don't make any computations. In fact, this layer provides the only way to pass user data into the network. As any other layer, this layer can label its outputs and this function provides an easy way to do this.
Sets the new value for the learned param of the layer.
layer | name or id of the layer. |
numParam | index of the layer parameter in the Layer::blobs array. |
blob | the new value. |