The class represents a builder for Detection Output layer. More...
#include <ie_detection_output_layer.hpp>
Public Member Functions | |
DetectionOutputLayer (const std::string &name="") | |
The constructor creates a builder with the name. More... | |
DetectionOutputLayer (const Layer::Ptr &layer) | |
The constructor creates a builder from generic builder. More... | |
DetectionOutputLayer (const Layer::CPtr &layer) | |
The constructor creates a builder from generic builder. More... | |
DetectionOutputLayer & | setName (const std::string &name) |
Sets the name for the layer. More... | |
const Port & | getOutputPort () const |
Returns output port. More... | |
DetectionOutputLayer & | setOutputPort (const Port &port) |
Sets output port. More... | |
const std::vector< Port > & | getInputPorts () const |
Returns input ports. More... | |
DetectionOutputLayer & | setInputPorts (const std::vector< Port > &ports) |
Sets input ports. More... | |
size_t | getNumClasses () const |
Returns number of classes. More... | |
DetectionOutputLayer & | setNumClasses (size_t num) |
Sets number of classes to be predict. More... | |
int | getBackgroudLabelId () const |
Returns background label ID. More... | |
DetectionOutputLayer & | setBackgroudLabelId (int labelId) |
Sets background label ID. More... | |
int | getTopK () const |
Returns maximum number of results to be kept on NMS stage. More... | |
DetectionOutputLayer & | setTopK (int topK) |
Sets maximum number of results to be kept on NMS stage. More... | |
int | getKeepTopK () const |
Returns number of total boxes to be kept per image after NMS step. More... | |
DetectionOutputLayer & | setKeepTopK (int topK) |
Sets number of total boxes to be kept per image after NMS step. More... | |
int | getNumOrientClasses () const |
Returns number of oriented classes. More... | |
DetectionOutputLayer & | setNumOrientClasses (int numClasses) |
Sets number of oriented classes. More... | |
std::string | getCodeType () const |
Returns type of coding method for bounding boxes. More... | |
DetectionOutputLayer & | setCodeType (std::string type) |
Sets type of coding method for bounding boxes. More... | |
int | getInterpolateOrientation () const |
Returns interpolate orientation. More... | |
DetectionOutputLayer & | setInterpolateOrientation (int orient) |
Sets interpolate orientation. More... | |
float | getNMSThreshold () const |
Returns threshold to be used in NMS stage. More... | |
DetectionOutputLayer & | setNMSThreshold (float threshold) |
Sets threshold to be used in NMS stage. More... | |
float | getConfidenceThreshold () const |
Returns confidence threshold. More... | |
DetectionOutputLayer & | setConfidenceThreshold (float threshold) |
Sets confidence threshold. More... | |
bool | getShareLocation () const |
Returns share location. More... | |
DetectionOutputLayer & | setShareLocation (bool flag) |
Sets share location. More... | |
bool | getVariantEncodedInTarget () const |
Returns encoded settings. More... | |
DetectionOutputLayer & | setVariantEncodedInTarget (bool flag) |
Sets encoded settings. 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 Detection Output 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 |
int InferenceEngine::Builder::DetectionOutputLayer::getBackgroudLabelId | ( | ) | const |
Returns background label ID.
std::string InferenceEngine::Builder::DetectionOutputLayer::getCodeType | ( | ) | const |
Returns type of coding method for bounding boxes.
float InferenceEngine::Builder::DetectionOutputLayer::getConfidenceThreshold | ( | ) | const |
Returns confidence threshold.
const std::vector<Port>& InferenceEngine::Builder::DetectionOutputLayer::getInputPorts | ( | ) | const |
Returns input ports.
int InferenceEngine::Builder::DetectionOutputLayer::getInterpolateOrientation | ( | ) | const |
Returns interpolate orientation.
int InferenceEngine::Builder::DetectionOutputLayer::getKeepTopK | ( | ) | const |
Returns number of total boxes to be kept per image after NMS step.
float InferenceEngine::Builder::DetectionOutputLayer::getNMSThreshold | ( | ) | const |
Returns threshold to be used in NMS stage.
size_t InferenceEngine::Builder::DetectionOutputLayer::getNumClasses | ( | ) | const |
Returns number of classes.
int InferenceEngine::Builder::DetectionOutputLayer::getNumOrientClasses | ( | ) | const |
Returns number of oriented classes.
const Port& InferenceEngine::Builder::DetectionOutputLayer::getOutputPort | ( | ) | const |
Returns output port.
bool InferenceEngine::Builder::DetectionOutputLayer::getShareLocation | ( | ) | const |
Returns share location.
int InferenceEngine::Builder::DetectionOutputLayer::getTopK | ( | ) | const |
Returns maximum number of results to be kept on NMS stage.
bool InferenceEngine::Builder::DetectionOutputLayer::getVariantEncodedInTarget | ( | ) | const |
Returns encoded settings.
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setBackgroudLabelId | ( | int | labelId | ) |
Sets background label ID.
labelId | Background ID if there is no background class, set it to -1. |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setCodeType | ( | std::string | type | ) |
Sets type of coding method for bounding boxes.
type | Type |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setConfidenceThreshold | ( | float | threshold | ) |
Sets confidence threshold.
threshold | Threshold |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setInputPorts | ( | const std::vector< Port > & | ports | ) |
Sets input ports.
ports | Vector of input ports |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setInterpolateOrientation | ( | int | orient | ) |
Sets interpolate orientation.
orient | Orientation |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setKeepTopK | ( | int | topK | ) |
Sets number of total boxes to be kept per image after NMS step.
topK | Keep top K |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setName | ( | const std::string & | name | ) |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setNMSThreshold | ( | float | threshold | ) |
Sets threshold to be used in NMS stage.
threshold | NMS threshold |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setNumClasses | ( | size_t | num | ) |
Sets number of classes to be predict.
num | Number of classes |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setNumOrientClasses | ( | int | numClasses | ) |
Sets number of oriented classes.
numClasses | Number of classes |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setOutputPort | ( | const Port & | port | ) |
Sets output port.
port | Output port |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setShareLocation | ( | bool | flag | ) |
Sets share location.
flag | true if bounding boxes are shared among different classes |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setTopK | ( | int | topK | ) |
Sets maximum number of results to be kept on NMS stage.
topK | Top K |
DetectionOutputLayer& InferenceEngine::Builder::DetectionOutputLayer::setVariantEncodedInTarget | ( | bool | flag | ) |
Sets encoded settings.
flag | true if variance is encoded in target |