Class ov::op::v0::PSROIPooling#

class PSROIPooling : public ov::op::Op#

PSROIPooling operation.

Public Functions

PSROIPooling(const Output<Node> &input, const Output<Node> &coords, const size_t output_dim, const size_t group_size, const float spatial_scale, int spatial_bins_x, int spatial_bins_y, const std::string &mode)#

Constructs a PSROIPooling operation.

Parameters:

  • input – Input feature map {N, C, …}

  • coords – Coordinates of bounding boxes

  • output_dim – Output channel number

  • group_size – Number of groups to encode position-sensitive scores

  • spatial_scale – Ratio of input feature map over input image size

  • spatial_bins_x – Numbers of bins to divide the input feature maps over width

  • spatial_bins_y – Numbers of bins to divide the input feature maps over height

  • mode – Mode of pooling - Avg or Bilinear

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.

void set_output_dim(size_t output_dim)#

Set the output channel dimension size.

Parameters:

output_dim – Channel dimension size.

void set_group_size(size_t group_size)#

Set the output groups number.

Parameters:

group_size – Number of groups.

void set_spatial_scale(float scale)#

Set the spatial scale.

Parameters:

scale – Spatial scale value.

void set_spatial_bins_x(int x)#

Set the number of bins over image width.

Parameters:

x – Number of bins over width (x) axis.

void set_spatial_bins_y(int y)#

Set the number of bins over image height.

Parameters:

y – Number of bins over height (y) axis.

void set_mode(std::string mode)#

Set the pooling mode.

Parameters:

mode – Pooling mode name.