Method RoiAlign

RoiAlign(TensorFloat, TensorFloat, TensorInt, TensorFloat, RoiPoolingMode, int, int, int, float)

Calculates an output tensor by pooling the input tensor across each region of interest given by the rois tensor.

Declaration
public override void RoiAlign(TensorFloat X, TensorFloat rois, TensorInt indices, TensorFloat O, RoiPoolingMode mode, int outputHeight, int outputWidth, int samplingRatio, float spatialScale)
Parameters
Type Name Description
TensorFloat X

The input tensor.
TensorFloat rois

The region of interest input tensor.
TensorInt indices

The indices input tensor.
TensorFloat O

The output tensor to be computed and filled.
RoiPoolingMode mode

The pooling mode of the operation as an RoiPoolingMode.
int outputHeight

The height of the output tensor.
int outputWidth

The width of the output tensor.
int samplingRatio

The number of sampling points in the interpolation grid used to compute the output value of each pooled output bin.
float spatialScale

The multiplicative spatial scale factor used to translate coordinates from their input spatial scale to the scale used when pooling.
Overrides