Ubuntu18.04——安装并测试GPU版本mmdet

文章目录

• ​​环境准备​​
• ​​安装mmdet并测试​​

• ​​安装​​
• ​​测试​​

环境准备

• ​​安装Anaconda和python3.6.9版本的pytorch-gpu​​

安装mmdet并测试

安装

建议安装MMDetection和MIM，它会自动安装OpenMMLab项目的依赖，包括mmcv等python包。

pip install openmim
mim install mmdet

测试

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

下载测试模型：

mkdir checkpoints
cd checkpoints
# 下载测试模型
wget https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

预测：

cd ..
# 预测
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

测试结果：

成功跑通～

参考文章：

• ​​MMdetection Get Started​​
• ​​MMDet安装过程记录​​