7 #include <builders/ie_layer_fragment.hpp>
18 class INFERENCE_ENGINE_API_CLASS(ProposalLayer):
public LayerFragment {
24 explicit ProposalLayer(
const std::string& name =
"");
29 explicit ProposalLayer(Layer& genLayer);
35 ProposalLayer& setName(
const std::string& name);
41 const Port& getOutputPort()
const;
47 ProposalLayer& setOutputPort(
const Port& port);
52 const std::vector<Port>& getInputPorts()
const;
58 ProposalLayer& setInputPorts(
const std::vector<Port>& ports);
63 size_t getPostNMSTopN()
const;
69 ProposalLayer& setPostNMSTopN(
size_t topN);
74 size_t getPreNMSTopN()
const;
80 ProposalLayer& setPreNMSTopN(
size_t topN);
85 float getNMSThresh()
const;
91 ProposalLayer& setNMSThresh(
float thresh);
96 size_t getBaseSize()
const;
102 ProposalLayer& setBaseSize(
size_t baseSize);
107 size_t getMinSize()
const;
113 ProposalLayer& setMinSize(
size_t minSize);
118 size_t getFeatStride()
const;
124 ProposalLayer& setFeatStride(
size_t featStride);
129 const std::vector<float> getScale()
const;
135 ProposalLayer& setScale(
const std::vector<float>& scales);
140 const std::vector<float> getRatio()
const;
146 ProposalLayer& setRatio(
const std::vector<float>& ratios);
Definition: ie_argmax_layer.hpp:11
a header file for the Inference Engine Network interface