Class ROIPooling
Defined in File roi_pooling.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class ROIPooling : public mindspore::ops::BaseOperator
ROIPooling defined the ROIPooling operator prototype.
Public Functions
-
inline ROIPooling()
Constructor.
-
void Init(const int64_t pooled_h, const int64_t pooled_w, const float scale)
Method to init the op’s attributes.
- Parameters
pooled_h – [in] Define the height of the output.
pooled_w – [in] Define the width of the output.
scale – [in] Define the size factor to translate ROI coordinates from the input coordinate to the actual coordinate.
-
void set_pooled_h(const int64_t pooled_h)
Method to set pooled_h attribute.
- Parameters
pooled_h – [in] Define the height of the output.
-
void set_pooled_w(const int64_t pooled_w)
Method to set pooled_w attribute.
- Parameters
pooled_w – [in] Define the width of the output.
-
void set_scale(const float scale)
Method to set scale attribute.
- Parameters
scale – [in] Define the size factor to translate ROI coordinates from the input coordinate to the actual coordinate.
-
int64_t get_pooled_h() const
Method to get pooled_h attribute.
- Returns
the height of the output.
-
int64_t get_pooled_w() const
Method to get pooled_w attribute.
- Returns
the width of the output.
-
float get_scale() const
Method to get scale attribute.
- Returns
the size factor.
-
inline ROIPooling()