19 #include "ngraph/op/op.hpp"
28 int background_label_id = 0;
30 bool variance_encoded_in_target =
false;
31 std::vector<int> keep_top_k;
32 std::string code_type = std::string{
"caffe.PriorBoxParameter.CORNER"};
33 bool share_location =
true;
35 float confidence_threshold = 0;
36 bool clip_after_nms =
false;
37 bool clip_before_nms =
false;
38 bool decrease_label_id =
false;
39 bool normalized =
false;
40 size_t input_height = 1;
41 size_t input_width = 1;
42 float objectness_score = 0;
52 static constexpr
NodeTypeInfo type_info{
"DetectionOutput", 0};
83 virtual std::shared_ptr<Node>
84 clone_with_new_inputs(
const OutputVector& new_args)
const override;
93 using v0::DetectionOutput;
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
Layer which performs non-max suppression to generate detection output using location and confidence p...
Definition: detection_output.hpp:50
DetectionOutput(const Output< Node > &box_logits, const Output< Node > &class_preds, const Output< Node > &proposals, const Output< Node > &aux_class_preds, const Output< Node > &aux_box_preds, const DetectionOutputAttrs &attrs)
Constructs a DetectionOutput operation.
DetectionOutput(const Output< Node > &box_logits, const Output< Node > &class_preds, const Output< Node > &proposals, const DetectionOutputAttrs &attrs)
Constructs a DetectionOutput operation.
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: detection_output.hpp:53
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Definition: detection_output.hpp:26