利用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 .

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值