【MMDetection-学习记录】 Windows10操作系统下安装并运行

参考文档:https://mmdetection.readthedocs.io/en/latest/get_started.html

 

安装过程中,版本需要对应,否则会出现问题。

MMDetection和MMVC对应版本如下:(使用MMDetection2.7.0 mmcv-full1.1.5)

https://github.com/open-mmlab/mmdetection/releases/tag/v2.7.0

MMDetection version MMCV version
master mmcv-full>=1.2.4, <1.3
2.8.0 mmcv-full>=1.2.4, <1.3
2.7.0 mmcv-full>=1.1.5, <1.3
2.6.0 mmcv-full>=1.1.5, <1.3
2.5.0 mmcv-full>=1.1.5, <1.3
2.4.0 mmcv-full>=1.1.1, <1.3
2.3.0 mmcv-full==1.0.5
2.3.0rc0 mmcv-full>=1.0.2
2.2.1 mmcv==0.6.2
2.2.0 mmcv==0.6.2
2.1.0 mmcv>=0.5.9, <=0.6.1
2.0.0 mmcv>=0.5.1, <=0.5.8

使用conda 创建虚拟环境并激活虚拟环境。

conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab

安装pytorch

Pytorch官网:https://pytorch.org/get-started/locally/

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

安装mmcv-full:

pip install mmcv-full==1.1.5 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html

Clone the MMDetection repository:

github地址:https://github.com/open-mmlab/mmdetection(可以进入github选择合适的版本下载)

mmdetection2.7.0(含有faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth)   百度网盘:https://pan.baidu.com/s/1stnGtsRZb4WrB2qws3jFhQ 提取码:etu7

下面的命令可以忽略: 

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection

安装MMDetection

pip install -r requirements/build.txt
python setup.py develop

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第1张图片

 

安装成功!!

如果安装过程出现下面问题:

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第2张图片

则在setup中添加一行代码:

os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第3张图片

mmdetection 文件夹下 demo文件夹下含有测试代码:

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第4张图片

还需要安装pycocotools(需要vc环境,可以参考博客:https://www.jianshu.com/p/8658cda3d553)、scipy、matplotlib(使用pip安装即可)

pip install mmpycocotools(安装mmpycocotools)

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第5张图片

https://github.com/philferriere/cocoapi下载源码,并进行解压。

以管理员身份打开 CMD 终端,并切换到 cocoapi\PythonAPI目录。运行以下指令:

python setup.py build_ext install

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第6张图片

 

测试是否安装成功:

下载checkpoints文件,并放到checkpoints文件夹下:

# download the checkpoint from model zoo and put it in `checkpoints/`

# url: http://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第7张图片

 

mmdetection-2.7.0文件夹下demo中含有测试代码:(包含图片检测 image_demo.py 和调用摄像头检测 webcam_demo.py

image_demo.py使用命令运行:

python demo\image_demo.py demo\demo.jpg configs\faster_rcnn\faster_rcnn_r50_fpn_1x_coco.py checkpoints\faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

 

测试结果:

换张图片试一下:

 

 

webcam_demo.py 使用命令运行:

python demo\webcam_demo.py configs\faster_rcnn\faster_rcnn_r50_fpn_1x_coco.py checkpoints\faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

 

运行结果:

【MMDetection-学习记录】 Windows10操作系统下安装并运行_第8张图片

 

代码结构图:

下图来源:https://www.cnblogs.com/wjy-lulu/p/13268321.html(如有侵权,联系删除!)

 

你可能感兴趣的:(pyTorch,mmdetection)