MMCV

功能简介

MMCV是一个主要用于计算机视觉研究的第三方库。

安装方法

当前昇腾适配的版本要求MMCV版本为1.7.0及以上。

用户可以选择MIM直接安装（推荐）或编译安装方式在昇腾设备上安装MMCV。

• MIM直接安装（推荐）。
  

  MIM提供了一个统一的接口，用于启动和安装open-mmlab项目及其扩展，以及管理open-mmlab模型库。

  执行如下命令安装MIM：

  pip install -U openmim

  执行如下命令安装MMCV。

  mim install mmcv

• 编译安装MMCV。
  • 拉取MMCV 源码。

    git pull https://github.com/open-mmlab/mmcv/tree/master

  • 编译。

    MMCV_WITH_OPS=1 MAX_JOBS=8 FORCE_NPU=1 python setup.py build_ext

  • 安装。

    MMCV_WITH_OPS=1 FORCE_NPU=1 python setup.py develop

支持算子

当前MMCV在昇腾设备上支持的算子列表请参考表1。完整的MMCV算子列表请参考MMCV算子列表。

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