7 #include <builders/ie_layer_fragment.hpp>
18 class INFERENCE_ENGINE_API_CLASS(RegionYoloLayer):
public LayerFragment {
24 explicit RegionYoloLayer(
const std::string& name =
"");
29 explicit RegionYoloLayer(Layer& genLayer);
35 RegionYoloLayer& setName(
const std::string& name);
41 const Port& getInputPort()
const;
47 RegionYoloLayer& setInputPort(
const Port& port);
52 const Port& getOutputPort()
const;
58 RegionYoloLayer& setOutputPort(
const Port& port);
63 int getCoords()
const;
69 RegionYoloLayer& setCoords(
int coords);
74 int getClasses()
const;
80 RegionYoloLayer& setClasses(
int classes);
91 RegionYoloLayer& setNum(
int num);
96 bool getDoSoftMax()
const;
102 RegionYoloLayer& setDoSoftMax(
bool flag);
107 float getAnchors()
const;
113 RegionYoloLayer& setAnchors(
float anchors);
124 RegionYoloLayer& setMask(
int mask);
129 size_t getAxis()
const;
135 RegionYoloLayer& setAxis(
size_t axis);
140 size_t getEndAxis()
const;
146 RegionYoloLayer& setEndAxis(
size_t axis);
Definition: ie_argmax_layer.hpp:11
a header file for the Inference Engine Network interface