MMDetection训练自己的数据集

最新推荐文章于 2024-06-26 18:12:58 发布

明素

最新推荐文章于 2024-06-26 18:12:58 发布

阅读量545

点赞数

分类专栏：目标检测文章标签：神经网络 pytorch

本文链接：https://blog.csdn.net/weixin_43406381/article/details/127820004

版权

本文档介绍了如何在MMDetection框架下，分别在Windows和Linux环境下安装，并详细说明了以COCO格式训练自己数据集的步骤，包括训练命令的使用和配置文件的修改。

摘要由CSDN通过智能技术生成

MMDetection

1. 安装

1.1 windows

1.1.1 克隆git仓库

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

1.1.2 创建环境

conda create --name openmmlab python=3.8 -y
conda activate openmmlab
conda install pytorch torchvision -c pytorch
pip install -U openmim
mim install mmcv-full
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
# "-v" means verbose, or more output
# "-e" means installing a project in editable mode,
# thus any local modifications made to the code will take effect without reinstallation.

1.1.3 验证

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