The class represents a builder for RegionYolo layer. More...
#include <ie_region_yolo_layer.hpp>
Public Member Functions | |
RegionYoloLayer (const std::string &name="") | |
The constructor creates a builder with the name. More... | |
RegionYoloLayer (const Layer::Ptr &layer) | |
The constructor creates a builder from generic builder. More... | |
RegionYoloLayer (const Layer::CPtr &layer) | |
The constructor creates a builder from generic builder. More... | |
RegionYoloLayer & | setName (const std::string &name) |
Sets the name for the layer. More... | |
const Port & | getInputPort () const |
Returns input port. More... | |
RegionYoloLayer & | setInputPort (const Port &port) |
Sets input port. More... | |
const Port & | getOutputPort () const |
Returns output port. More... | |
RegionYoloLayer & | setOutputPort (const Port &port) |
Sets output port. More... | |
int | getCoords () const |
Returns number of coordinates for each region. More... | |
RegionYoloLayer & | setCoords (int coords) |
Sets number of coordinates for each region. More... | |
int | getClasses () const |
Returns number of classes for each region. More... | |
RegionYoloLayer & | setClasses (int classes) |
Sets number of classes for each region. More... | |
int | getNum () const |
Returns number of regions. More... | |
RegionYoloLayer & | setNum (int num) |
Sets number of regions. More... | |
bool | getDoSoftMax () const |
Returns a flag which specifies the method of infer. More... | |
RegionYoloLayer & | setDoSoftMax (bool flag) |
Sets a flag which specifies the method of infer. More... | |
float | getAnchors () const |
Returns anchors coordinates of regions. More... | |
RegionYoloLayer & | setAnchors (float anchors) |
Sets anchors coordinates of regions. More... | |
int | getMask () const |
Returns mask. More... | |
RegionYoloLayer & | setMask (int mask) |
Sets mask. More... | |
size_t | getAxis () const |
Returns the number of the dimension from which flattening is performed. More... | |
RegionYoloLayer & | setAxis (size_t axis) |
Sets the number of the dimension from which flattening is performed. More... | |
size_t | getEndAxis () const |
Returns the number of the dimension on which flattening is ended. More... | |
RegionYoloLayer & | setEndAxis (size_t axis) |
Sets the number of the dimension on which flattening is ended. More... | |
Public Member Functions inherited from InferenceEngine::Builder::LayerDecorator | |
LayerDecorator (const std::string &type, const std::string &name) | |
The constructor creates layer builders with layer type and layer name. More... | |
LayerDecorator (const Layer::Ptr &layer) | |
The constructor creates layer builders from reference to generic layer builder. More... | |
LayerDecorator (const Layer::CPtr &layer) | |
The constructor creates layer builders from reference to generic layer builder. More... | |
LayerDecorator (const LayerDecorator &rval) | |
The copy constructor. More... | |
LayerDecorator & | operator= (const LayerDecorator &rval) |
Copy operator for LayerDecorator. More... | |
virtual | ~LayerDecorator ()=default |
Virtual destructor. | |
virtual | operator Layer () const |
The operator creates generic builder. More... | |
virtual | operator Layer::Ptr () |
The operator creates generic builder. More... | |
virtual | operator Layer::CPtr () const |
The operator creates generic builder. More... | |
const std::string & | getType () const |
Returns layer type. More... | |
const std::string & | getName () const |
Returns layer name. More... | |
Additional Inherited Members | |
Protected Member Functions inherited from InferenceEngine::Builder::LayerDecorator | |
Layer::Ptr & | getLayer () |
const Layer::CPtr | getLayer () const |
void | checkType (const std::string &type) const |
Protected Attributes inherited from InferenceEngine::Builder::LayerDecorator | |
Layer::CPtr | cLayer |
The class represents a builder for RegionYolo layer.
|
explicit |
The constructor creates a builder with the name.
name | Layer name |
|
explicit |
The constructor creates a builder from generic builder.
layer | pointer to generic builder |
|
explicit |
The constructor creates a builder from generic builder.
layer | constant pointer to generic builder |
float InferenceEngine::Builder::RegionYoloLayer::getAnchors | ( | ) | const |
Returns anchors coordinates of regions.
size_t InferenceEngine::Builder::RegionYoloLayer::getAxis | ( | ) | const |
Returns the number of the dimension from which flattening is performed.
int InferenceEngine::Builder::RegionYoloLayer::getClasses | ( | ) | const |
Returns number of classes for each region.
int InferenceEngine::Builder::RegionYoloLayer::getCoords | ( | ) | const |
Returns number of coordinates for each region.
bool InferenceEngine::Builder::RegionYoloLayer::getDoSoftMax | ( | ) | const |
Returns a flag which specifies the method of infer.
size_t InferenceEngine::Builder::RegionYoloLayer::getEndAxis | ( | ) | const |
Returns the number of the dimension on which flattening is ended.
const Port& InferenceEngine::Builder::RegionYoloLayer::getInputPort | ( | ) | const |
Returns input port.
int InferenceEngine::Builder::RegionYoloLayer::getMask | ( | ) | const |
Returns mask.
int InferenceEngine::Builder::RegionYoloLayer::getNum | ( | ) | const |
Returns number of regions.
const Port& InferenceEngine::Builder::RegionYoloLayer::getOutputPort | ( | ) | const |
Returns output port.
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setAnchors | ( | float | anchors | ) |
Sets anchors coordinates of regions.
anchors | Anchors coordinates |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setAxis | ( | size_t | axis | ) |
Sets the number of the dimension from which flattening is performed.
axis | Axis |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setClasses | ( | int | classes | ) |
Sets number of classes for each region.
classes | number of classes |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setCoords | ( | int | coords | ) |
Sets number of coordinates for each region.
coords | Number of coordinates |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setDoSoftMax | ( | bool | flag | ) |
Sets a flag which specifies the method of infer.
flag | softmax is performed if true |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setEndAxis | ( | size_t | axis | ) |
Sets the number of the dimension on which flattening is ended.
axis | End axis |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setInputPort | ( | const Port & | port | ) |
Sets input port.
port | Input port |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setMask | ( | int | mask | ) |
Sets mask.
mask | Specifies which anchors to use |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setName | ( | const std::string & | name | ) |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setNum | ( | int | num | ) |
Sets number of regions.
num | Number of regions |
RegionYoloLayer& InferenceEngine::Builder::RegionYoloLayer::setOutputPort | ( | const Port & | port | ) |
Sets output port.
port | Output port |