RoiAlign是一种池化层,用于非均匀输入尺寸的特征图,并输出固定尺寸的特征图。对标ONNX opset 10算子原型,aclnnRoiAlignV2对标torchvision算子原型。aclnnRoiAlignV2使用boxes替代aclnnRoiAlign的rois和batch_indices,并增加aligned入参,同时取消mode入参、默认执行mode="avg"场景。
每个算子分为,必须先调用“aclnnRoiAlignV2GetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnRoiAlignV2”接口执行计算。
[object Object]
[object Object]
- 确定性计算:
- aclnnRoiAlignV2默认确定性实现。
[object Object]