RoiAlign is a pooling layer used to input a feature map with a non-uniform size and output a fixed-size feature map. corresponds to the ONNX opset 10 operator prototype, and aclnnRoiAlignV2 corresponds to the torchvision operator prototype. aclnnRoiAlignV2 uses boxes to replace rois and batch_indices of aclnnRoiAlign, adds the aligned input parameter, and removes the mode argument, using mode="avg" by default.
Each operator has calls. First, aclnnRoiAlignV2GetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, aclnnRoiAlignV2 is called to perform computation.
Parameters
[object Object]Returns:
[object Object]: status code. For details, see .The first-phase API implements input parameter validation. The following error codes may be returned.
[object Object]
- Deterministic computation:
- aclnnRoiAlignV2 defaults to a deterministic implementation.
The following example is for reference only. For details, see .