7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
18 class INFERENCE_ENGINE_API_CLASS(SimplerNMSLayer):
public LayerDecorator {
24 explicit SimplerNMSLayer(
const std::string& name =
"");
29 explicit SimplerNMSLayer(
const Layer::Ptr& layer);
34 explicit SimplerNMSLayer(
const Layer::CPtr& layer);
40 SimplerNMSLayer& setName(
const std::string& name);
46 const std::vector<Port>& getInputPorts()
const;
51 SimplerNMSLayer& setInputPorts(
const std::vector<Port>& ports);
56 const Port& getOutputPort()
const;
62 SimplerNMSLayer& setOutputPort(
const Port& port);
67 size_t getPreNMSTopN()
const;
73 SimplerNMSLayer& setPreNMSTopN(
size_t topN);
78 size_t getPostNMSTopN()
const;
84 SimplerNMSLayer& setPostNMSTopN(
size_t topN);
89 size_t getFeatStride()
const;
95 SimplerNMSLayer& setFeatStride(
size_t featStride);
100 size_t getMinBoxSize()
const;
106 SimplerNMSLayer& setMinBoxSize(
size_t minSize);
111 size_t getScale()
const;
117 SimplerNMSLayer& setScale(
size_t scale);
123 float getCLSThreshold()
const;
129 SimplerNMSLayer& setCLSThreshold(
float threshold);
134 float getIOUThreshold()
const;
140 SimplerNMSLayer& setIOUThreshold(
float threshold);
Definition: ie_argmax_layer.hpp:11