7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
18 class INFERENCE_ENGINE_API_CLASS(DetectionOutputLayer):
public LayerDecorator {
24 explicit DetectionOutputLayer(
const std::string& name =
"");
29 explicit DetectionOutputLayer(
const Layer::Ptr& layer);
34 explicit DetectionOutputLayer(
const Layer::CPtr& layer);
40 DetectionOutputLayer& setName(
const std::string& name);
46 const Port& getOutputPort()
const;
52 DetectionOutputLayer& setOutputPort(
const Port& port);
57 const std::vector<Port>& getInputPorts()
const;
63 DetectionOutputLayer& setInputPorts(
const std::vector<Port>& ports);
68 size_t getNumClasses()
const;
74 DetectionOutputLayer& setNumClasses(
size_t num);
79 int getBackgroudLabelId()
const;
85 DetectionOutputLayer& setBackgroudLabelId(
int labelId);
96 DetectionOutputLayer& setTopK(
int topK);
101 int getKeepTopK()
const;
107 DetectionOutputLayer& setKeepTopK(
int topK);
112 int getNumOrientClasses()
const;
118 DetectionOutputLayer& setNumOrientClasses(
int numClasses);
123 std::string getCodeType()
const;
129 DetectionOutputLayer& setCodeType(std::string type);
134 int getInterpolateOrientation()
const;
140 DetectionOutputLayer& setInterpolateOrientation(
int orient);
145 float getNMSThreshold()
const;
151 DetectionOutputLayer& setNMSThreshold(
float threshold);
156 float getConfidenceThreshold()
const;
162 DetectionOutputLayer& setConfidenceThreshold(
float threshold);
167 bool getShareLocation()
const;
173 DetectionOutputLayer& setShareLocation(
bool flag);
178 bool getVariantEncodedInTarget()
const;
184 DetectionOutputLayer& setVariantEncodedInTarget(
bool flag);
Definition: ie_argmax_layer.hpp:11