跳转至

MMCV

MMCV 是一个面向计算机视觉的基础库,它支持了很多开源项目。

警告

RTX 3000 系列显卡在 PyTorch 1.8.1 (For CUDA 11.1) 上 MMCV 目前工作不正常,使用此系列显卡请创建 PyTorch 1.9.1 版本镜像。PyTorch 1.10 版本暂时未安装 MMCV。

PyTorch 官方镜像已经预装了 mmcv-full 的包,可以直接安装 OpenMMLab 其他的工具箱进行使用。

MMDetection

MMDetection 是一个基于 PyTorch 的目标检测开源工具箱。

MMDetection 对 MMCV 的版本要求查看 Prerequisites

# 查看 MMCV 版本
pip show mmcv-full | grep Version

git clone https://mirror.ghproxy.com/https://github.com/open-mmlab/mmdetection.git
cd mmdetection
# 根据 MMCV 的版本检出支持的 MMDetection 版本
git checkout v2.14.0

# 安装
pip install -r requirements/build.txt --no-cache
pip install -v -e . --no-cache

接下来下载一个模型文件用于测试。

mkdir checkpoints
cd checkpoints
curl -OL "http://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth"

打开 JupyterLab,打开刚刚克隆下代码中的 /root/mmdetection/demo/inference_demo.ipynb 记事本文件。点击菜单中的 运行 - 运行所有单元格。