昇腾社区首页
中文
注册

昇腾支持的MMCV算子列表

当前昇腾NPU支持的MMCV算子列表请参考表1。完整的MMCV算子列表请参考MMCV算子列表

  • 使用ActiveRotatedFilter算子时,参数indices的生成方式请参考mmrotate/orconv.py
  • 使用BBoxOverlaps算子时,不建议输入检测框宽和高的大小在10的8次方数量级及以上。超出这个限制会导致MMCV该算子的CPU实现出现计算错误,导致NPU计算结果与标杆对比不一致。
  • 使用MaskedConv算子时,参数bias默认为True,暂时不建议使用bias=False的情况。
表1 昇腾支持的MMCV算子支持列表

算子名称

CPU

CUDA

昇腾设备

ActiveRotatedFilter

-

BBoxOverlaps

-

BoxIouRotated

-

BoxIouQuadri

-

ContourExpand

-

GatherPoints

-

MaskedConv

-

NMS

NMSQuadri

-

PixelGroup

-

PointsInBoxes

-

PSAMask

RotatedFeatureAlign

-

RoIPool

-

RoIAlign