1 预处理
1.1 配置要求
MMCV(相匹配的版本如下所示)
MMCV:Compatible MMDetection and MMCV versions are shown as below. Please install the correct version of MMCV to avoid installation issues
Note: You need to run pip uninstall mmcv first if you have mmcv installed. If mmcv and mmcv-full are both installed, there will be ModuleNotFoundError.(如果原来有mmcv的话先把原来的删掉)
1.2 查看指令
2 安装步骤
2.1 新建环境
指令
① conda create -n open-mmlab python=3.7 -y #新建虚拟环境“open-mmlab” ② conda activate open-mmlab # 或 source activate open-mmlab #进入虚拟环境
2.2 安装pytorch
2.2.1 pytorch官网:Start Locally | PyTorch(官网查看对应版本)
2.2.2 相关指令(下面是博主自己对应的版本)
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
2.3 安装mmcv-full(新版一定要安装mmcv-full)
GitHub - open-mmlab/mmcv: OpenMMLab Computer Vision Foundationhttps://github.com/open-mmlab/mmcv#installation(官网查看对应的版本)
安装指令:(这也是博主自己的对应版本,不一定适用)
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
成功
2.4 安装MMDetection
2.4.1 可以使用以下命令简单地安装MMDetection
pip install mmdet
2.4.2 或者使用克隆存储库,然后安装
① git clone https://github.com/open-mmlab/mmdetection.git
这里可能会报错,通常由于服务器版本问题与无vpn连接GitHub速度缓慢造成。
建议改为:
git clone git://github.com/open-mmlab/mmdetection.git
(下载速度异常缓慢,建议听会儿歌)搞定!
② cd mmdetection ③ pip install -r requirements/build.txt
④ pip install -v -e . # or "python setup.py develop"(成功)
3 测试部分
3.1 新建文件夹
在mmdetection下新建checkpoints文件夹
指令:mkdir checkpoints
在mmdetection/configs目录下有很多个模型,选择其中一种模型的文件夹进去,找到README.md下载对应的模型文件,放到checkpoints文件夹内。这里选择mask_rcnn_r50_fpn_1x_coco_20200205-d4b0c5d6.pth
输入指令: python demo/image_demo.py demo/demo.jpg configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py checkpoints/mask_rcnn_r50_fpn_1x_coco_20200205-d4b0c5d6.pth(测试成功)
OK,大功告成,拜拜~



