7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
18 class INFERENCE_ENGINE_API_CLASS(RegionYoloLayer):
public LayerDecorator {
24 explicit RegionYoloLayer(
const std::string& name =
"");
29 explicit RegionYoloLayer(
const Layer::Ptr& layer);
34 explicit RegionYoloLayer(
const Layer::CPtr& layer);
40 RegionYoloLayer& setName(
const std::string& name);
46 const Port& getInputPort()
const;
52 RegionYoloLayer& setInputPort(
const Port& port);
57 const Port& getOutputPort()
const;
63 RegionYoloLayer& setOutputPort(
const Port& port);
68 int getCoords()
const;
74 RegionYoloLayer& setCoords(
int coords);
79 int getClasses()
const;
85 RegionYoloLayer& setClasses(
int classes);
96 RegionYoloLayer& setNum(
int num);
101 bool getDoSoftMax()
const;
107 RegionYoloLayer& setDoSoftMax(
bool flag);
112 float getAnchors()
const;
118 RegionYoloLayer& setAnchors(
float anchors);
129 RegionYoloLayer& setMask(
int mask);
134 size_t getAxis()
const;
140 RegionYoloLayer& setAxis(
size_t axis);
145 size_t getEndAxis()
const;
151 RegionYoloLayer& setEndAxis(
size_t axis);
Definition: ie_argmax_layer.hpp:11