Class ov::op::v0::RegionYolo

class RegionYolo : public ov::op::Op

RegionYolo operation.

Public Functions

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.

Parameters

  • input – [in] Input

  • coords – [in] Number of coordinates for each region

  • classes – [in] Number of classes for each region

  • regions – [in] Number of regions

  • do_softmax – [in] Compute softmax

  • mask – [in] Mask

  • axis – [in] Axis to begin softmax on

  • end_axis – [in] Axis to end softmax on

  • anchors – [in] A flattened list of pairs [width, height] that describes prior box sizes.

virtual void validate_and_infer_types() override

Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.

Throws if the node is invalid.