19 #include "ngraph/op/op.hpp"
30 static constexpr
NodeTypeInfo type_info{
"RegionYolo", 0};
52 const bool do_softmax,
53 const std::vector<int64_t>& mask,
56 const std::vector<float>& anchors = std::vector<float>{});
61 virtual std::shared_ptr<Node>
62 clone_with_new_inputs(
const OutputVector& new_args)
const override;
64 size_t get_num_coords()
const {
return m_num_coords; }
65 size_t get_num_classes()
const {
return m_num_classes; }
66 size_t get_num_regions()
const {
return m_num_regions; }
67 bool get_do_softmax()
const {
return m_do_softmax; }
68 const std::vector<int64_t>& get_mask()
const {
return m_mask; }
69 const std::vector<float>& get_anchors()
const {
return m_anchors; }
70 int get_axis()
const {
return m_axis; }
71 int get_end_axis()
const {
return m_end_axis; }
77 std::vector<int64_t> m_mask;
78 std::vector<float> m_anchors{};
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
Definition: region_yolo.hpp:28
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
const NodeTypeInfo & get_type_info() const override
Definition: region_yolo.hpp:31
RegionYolo(const Output< Node > &input, const size_t coords, const size_t classes, const size_t regions, const bool do_softmax, const std::vector< int64_t > &mask, const int axis, const int end_axis, const std::vector< float > &anchors=std::vector< float >{})
Constructs a RegionYolo operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28