7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
19 IE_SUPPRESS_DEPRECATED_START
20 class INFERENCE_ENGINE_NN_BUILDER_API_CLASS(DetectionOutputLayer):
public LayerDecorator {
26 explicit DetectionOutputLayer(
const std::string& name =
"");
31 explicit DetectionOutputLayer(
const Layer::Ptr& layer);
36 explicit DetectionOutputLayer(
const Layer::CPtr& layer);
42 DetectionOutputLayer& setName(
const std::string& name);
48 const Port& getOutputPort()
const;
54 DetectionOutputLayer& setOutputPort(
const Port& port);
59 const std::vector<Port>& getInputPorts()
const;
65 DetectionOutputLayer& setInputPorts(
const std::vector<Port>& ports);
70 size_t getNumClasses()
const;
76 DetectionOutputLayer& setNumClasses(
size_t num);
81 int getBackgroudLabelId()
const;
87 DetectionOutputLayer& setBackgroudLabelId(
int labelId);
98 DetectionOutputLayer& setTopK(
int topK);
103 int getKeepTopK()
const;
109 DetectionOutputLayer& setKeepTopK(
int topK);
114 int getNumOrientClasses()
const;
120 DetectionOutputLayer& setNumOrientClasses(
int numClasses);
125 std::string getCodeType()
const;
131 DetectionOutputLayer& setCodeType(std::string type);
136 int getInterpolateOrientation()
const;
142 DetectionOutputLayer& setInterpolateOrientation(
int orient);
147 float getNMSThreshold()
const;
153 DetectionOutputLayer& setNMSThreshold(
float threshold);
158 float getConfidenceThreshold()
const;
164 DetectionOutputLayer& setConfidenceThreshold(
float threshold);
169 bool getShareLocation()
const;
175 DetectionOutputLayer& setShareLocation(
bool flag);
180 bool getVariantEncodedInTarget()
const;
186 DetectionOutputLayer& setVariantEncodedInTarget(
bool flag);
188 IE_SUPPRESS_DEPRECATED_END
Inference Engine API.
Definition: ie_argmax_layer.hpp:11