文章目录
- 前言
- 1、本地环境
- 2、安装
* - 2.1.步骤
- 2.2.运行demo
- 3、FasterRcnn训练coco2017数据集
* - 3.1.训练集准备
- 3.2.训练及测试Faster RCNN
- 总结
- 参考资料
前言
两个月没碰mmdetection了，今天突然发现更新迭代了好几版了，于是重新安装体验了下，发现安装及使用越来越方便了。故打算重新更新几篇mmdet的博文。
本文介绍最新的mmdet = 2.25.1版本的安装及demo及运行coco数据集训练。
1、本地环境
显卡 = 3090
CUDA=11.3
pytorch == 1.7.1
torchvision == 0.8.2
openmim == 0.2.0
mmcv-full == 1.6.1
mmdetection == 2.25.1
大家可以首先配置好pytorch和torchvision的虚拟环境。mmcv现在依靠openmim安装及运行。 当然open-mmlab系列的所有库均可使用open-mim直接安装，但本文站在开发角度，所以mmdet使用开发者模式安装。
2、安装
2.1.步骤
step 1：创建虚拟环境
conda create –name openmmlab python=3.7 -y
conda activate openmmlab
step 2：安装所需的pytorch和torchvision
在激活虚拟环境后，便可安装 mim 及 mmcv，其中mim会自动下载对应torch版本的mmcv，省事了不少。
pip install -U openmim
mim install mmcv-full
非常简单。之后便可安装mmdet：
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
编译成功后打印下我的pip list:
; 2.2.运行demo
新版的mmdet也用mim提供了一个简单易用的验证程序。废话不多说，上代码：
cd mmdetection
mim download mmdet --config yolov3_mobilenetv2_320_300e_coco --dest .
python demo/image_demo.py demo/demo.jpg yolov3_mobilenetv2_320_300e_coco.py yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth --device cpu --out-file result.jpg
运行完代码后，会在mmdetection目录下出现一张result.jpg的图像：
如果出现上述图像，说明安装成功了。
3、FasterRcnn训练coco2017数据集
3.1.训练集准备
首先下载个coco数据集，如果下载比较费劲，欢迎跳转我旧版mmdet安装教程，里面提供了一个voc版转coco数据集的教程，方便大家训练。以下是个文件树，按照以下格式准备自己数据集。
mmdetection
├── mmdet
├── tools
├── configs
├── data
│ ├── coco
│ │ ├── annotations
│ │ ├── train2017
│ │ ├── val2017
│ │ ├── test2017
3.2.训练及测试Faster RCNN
训练命令：
python tools/train.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py
训练效果图：
测试命令：官方github提供了多种测试命令，具体可见参考链接:
python tools/test.py \
configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py \
checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth \
--show-dir faster_rcnn_r50_fpn_1x_results
总结
mmdetection的安装更加人性化了，但mmdetection的整体架构没有变，因此，该系列文章会更加详细介绍mmdetection的架构流程以及检测论文代码解读等，敬请期待。若有问题欢迎+vx：wulele2541612007，加群一起交流学习。
参考资料
1、mmdetection的install文档
2、mmdetection的测试文档
Original: https://blog.csdn.net/wulele2/article/details/126200257
Author: 武乐乐~
Title: mmdetection最新版食用教程（一）：安装并运行demo及开始训练coco
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/712262/
转载文章受原作者版权保护。转载请注明原作者出处！