Class NonMaxSuppression

• Defined in File non_max_suppression.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class NonMaxSuppression : public mindspore::ops::BaseOperator

NonMaxSuppression QuantDTypeCast the NonMaxSuppression operator prototype.

Public Functions

inline NonMaxSuppression()

Constructor.

void Init(const int64_t center_point_box = 0)

Method to init the op’s attributes.

Parameters

center_point_box – [in] Define a value to indicate the format of the box data. If the value is 0, the box data is supplied by diagonal point, such as [y1, x1, y2, x2](the pair [y1, x1] is picture coordinate). If the value is 1, the box data is supplied as [x_center, y_center, width, height].

void set_center_point_box(const int64_t center_point_box)

Method to set center_point_box attribute.

Parameters

center_point_box – [in] a value to indicate the format of the box data.

int64_t get_center_point_box() const

Method to get center_point_box attribute.

Returns

an integer value.