利用mmdetection训练自己的VOC格式数据

最近呢,领导让我研究除了YOLO系列还有没有其他好用的小目标检测方法

就一顿百度搜索,打开N多个网页一一试错...特此记录,下面的链接都是亲身实践过可以成功运行的,不想自己写了,下次直接follow

mmdetection官方github:mmdetection/README_zh-CN.md at master · open-mmlab/mmdetection · GitHub

FasterRcnn:

MMDetection v2.0 训练自己的数据集

最新版本的mmdetection2.0 (v2.0.0版本)环境搭建、训练自己的数据集、测试以及常见错误集合

cascade-rcnn:

使用mmdetction-cascade_rcnn用自己数据训练并测试

mmdetection安装步骤 :

conda create -n openmmlab python=3.7 -y
conda activate openmmlab

#官方
#conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch -y

#本人
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html

# install the latest mmcv 注意跟自己的cuda和Pytorch版本匹配
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html

# install mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .

猜你喜欢

转载自blog.csdn.net/qq_36076110/article/details/120269440
今日推荐