7 #include <builders/ie_layer_fragment.hpp>
18 class INFERENCE_ENGINE_API_CLASS(SimplerNMSLayer):
public LayerFragment {
24 explicit SimplerNMSLayer(
const std::string& name =
"");
29 explicit SimplerNMSLayer(Layer& genLayer);
35 SimplerNMSLayer& setName(
const std::string& name);
41 const std::vector<Port>& getInputPorts()
const;
46 SimplerNMSLayer& setInputPorts(
const std::vector<Port>& ports);
51 const Port& getOutputPort()
const;
57 SimplerNMSLayer& setOutputPort(
const Port& port);
62 size_t getPreNMSTopN()
const;
68 SimplerNMSLayer& setPreNMSTopN(
size_t topN);
73 size_t getPostNMSTopN()
const;
79 SimplerNMSLayer& setPostNMSTopN(
size_t topN);
84 size_t getFeatStride()
const;
90 SimplerNMSLayer& setFeatStride(
size_t featStride);
95 size_t getMinBoxSize()
const;
101 SimplerNMSLayer& setMinBoxSize(
size_t minSize);
106 size_t getScale()
const;
112 SimplerNMSLayer& setScale(
size_t scale);
118 float getCLSThreshold()
const;
124 SimplerNMSLayer& setCLSThreshold(
float threshold);
129 float getIOUThreshold()
const;
135 SimplerNMSLayer& setIOUThreshold(
float threshold);
Definition: ie_argmax_layer.hpp:11
a header file for the Inference Engine Network interface