7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
19 IE_SUPPRESS_DEPRECATED_START
20 class INFERENCE_ENGINE_NN_BUILDER_API_CLASS(RegionYoloLayer):
public LayerDecorator {
26 explicit RegionYoloLayer(
const std::string& name =
"");
31 explicit RegionYoloLayer(
const Layer::Ptr& layer);
36 explicit RegionYoloLayer(
const Layer::CPtr& layer);
42 RegionYoloLayer& setName(
const std::string& name);
48 const Port& getInputPort()
const;
54 RegionYoloLayer& setInputPort(
const Port& port);
59 const Port& getOutputPort()
const;
65 RegionYoloLayer& setOutputPort(
const Port& port);
70 int getCoords()
const;
76 RegionYoloLayer& setCoords(
int coords);
81 int getClasses()
const;
87 RegionYoloLayer& setClasses(
int classes);
98 RegionYoloLayer& setNum(
int num);
103 bool getDoSoftMax()
const;
109 RegionYoloLayer& setDoSoftMax(
bool flag);
114 float getAnchors()
const;
120 RegionYoloLayer& setAnchors(
float anchors);
131 RegionYoloLayer& setMask(
int mask);
136 size_t getAxis()
const;
142 RegionYoloLayer& setAxis(
size_t axis);
147 size_t getEndAxis()
const;
153 RegionYoloLayer& setEndAxis(
size_t axis);
155 IE_SUPPRESS_DEPRECATED_END
Inference Engine API.
Definition: ie_argmax_layer.hpp:11