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

是木对啊

已于 2022-07-02 10:09:29 修改

阅读量763

点赞数

文章标签：深度学习

于 2021-09-13 16:29:34 首次发布

本文链接：https://blog.csdn.net/qq_36076110/article/details/120269440

版权

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

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

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

FasterRcnn：

MMDetection v2.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 .