7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
18 class INFERENCE_ENGINE_API_CLASS(ProposalLayer):
public LayerDecorator {
24 explicit ProposalLayer(
const std::string& name =
"");
29 explicit ProposalLayer(
const Layer::Ptr& layer);
34 explicit ProposalLayer(
const Layer::CPtr& layer);
40 ProposalLayer& setName(
const std::string& name);
46 const Port& getOutputPort()
const;
52 ProposalLayer& setOutputPort(
const Port& port);
57 const std::vector<Port>& getInputPorts()
const;
63 ProposalLayer& setInputPorts(
const std::vector<Port>& ports);
68 size_t getPostNMSTopN()
const;
74 ProposalLayer& setPostNMSTopN(
size_t topN);
79 size_t getPreNMSTopN()
const;
85 ProposalLayer& setPreNMSTopN(
size_t topN);
90 float getNMSThresh()
const;
96 ProposalLayer& setNMSThresh(
float thresh);
101 size_t getBaseSize()
const;
107 ProposalLayer& setBaseSize(
size_t baseSize);
112 size_t getMinSize()
const;
118 ProposalLayer& setMinSize(
size_t minSize);
123 size_t getFeatStride()
const;
129 ProposalLayer& setFeatStride(
size_t featStride);
134 const std::vector<float> getScale()
const;
140 ProposalLayer& setScale(
const std::vector<float>& scales);
145 const std::vector<float> getRatio()
const;
151 ProposalLayer& setRatio(
const std::vector<float>& ratios);
Definition: ie_argmax_layer.hpp:11