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 = {1};
32 std::string code_type = std::string{
"caffe.PriorBoxParameter.CORNER"};
33 bool share_location =
true;
35 float confidence_threshold = std::numeric_limits<float>::min();
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};
53 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
64 const Output<Node>& class_preds,
65 const Output<Node>& proposals,
66 const Output<Node>& aux_class_preds,
67 const Output<Node>& aux_box_preds,
77 const Output<Node>& class_preds,
78 const Output<Node>& proposals,
83 virtual std::shared_ptr<Node>
84 clone_with_new_inputs(
const OutputVector& new_args)
const override;
93 using v0::DetectionOutput;
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 DetectionOutputAttrs &attrs)
Constructs a DetectionOutput operation.
const NodeTypeInfo & get_type_info() const override
Definition: detection_output.hpp:53
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Definition: detection_output.hpp:26
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.
Root of all actual ops.
Definition: op.hpp:29