云服务器上安装MMDetection及训练自己的数据集-方法总结

1.安装mmcv和mmdetection

第一步:使用清华源来安装pytorch、torchvision和cython

直接安装pytorch和torchvision,命令如下:

install pytorch torchvision cudatoolkit=10.0
install cython

这一步由于我使用的是云服务器,机器已经把基础环境配置好了,我用的版本是:
pytorch1.6.0
CUDA10.2
python 3.6
登陆工具为XFTP,Xshell
注意:版本一定要对应(包括后续安装),避免安装过程中各种莫名其妙的错误!!!

第二步:git下载mmcv和mmdetection:

从github上下载mmcv和mmdetection,可能会导致超时或过慢:(不建议)

git clone https://github.com/open-mmlab/mmcv.git
git clone https://github.com/open-mmlab/mmdetection.git
1.从码云上下载mmcv和mmdetection的代码到云主机,命令如下:(建议使用
git clone https://gitee.com/mirrors/mmcv.git
git clone https://gitee.com/mirrors/mmdetection.git
2.确保一些必要的包已经安装:

不要提前安装pycocotools!!!因为mmdetection是使用的自己开发的pycocotools,当系统中已经安装了此包的时候,mmdetection不会覆盖安装,会导致后面安装失败!!(找了好久才解决)

pip install pytest-runner 
pip install pytest 
pip install addict
pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install pyyaml
pip install yaps   #这个很可能需要安装
pip install matplotlib  terminaltables   imagecorruptions  albumentations
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple matplotlib==3.3.1
3.安装mmcv-full

mmcv-full为GPU版本,mmcv为CPU版本,官方提示两者不可同时安装。

pip install mmcv-full==1.2.4 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0
  • 8
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
### 回答1: MMDetection是一个开源目标检测框架,它支持许多不同的目标检测模型,包括Faster R-CNN。要训练Faster R-CNN模型,您需要准备数据集、模型配置文件以及训练脚本。可以参考MMDetection的官方文档(https://mmdetection.readthedocs.io/en/latest/tutorials/train.html),了解详细的训练步骤。 ### 回答2: mmdetection是一个用于目标检测的开源框架。其中,Faster R-CNN是其中一种比较常用的检测网络,主要通过Region Proposal Network(RPN)生成候选区域,然后利用RoI Pooling对这些区域进行特征提取和分类。在训练Faster R-CNN时,我们需要做以下几步: 1. 数据预处理:将数据转换为合适的格式,例如将图片和标注转换为COCO(Common Objects in Context)格式。 2. 模型搭建:使用mmdetection中提供的配置文件进行模型搭建,并进行相应的超参数设置。在进行设置时,需要考虑到训练时可能需要使用的GPU数量,以及学习率,权重衰减等参数。 3. 数据扩增:为了提高模型的鲁棒性,在训练前需要进行数据扩增,常用的方法包括随机裁剪、随机翻转、随机旋转等。 4. 数据加载:定义DataLoader来实现数据的批量加载,并设置数据的并行数,以便加快训练速度。 5. 损失函数:Faster R-CNN中常用的损失函数包括分类损失和回归损失。在训练时需要同时计算这两种损失函数,并设置相应的权重。 6. 模型训练:对模型进行训练。在训练过程中,可以使用Adam等优化器对模型进行优化,并进行学习率调整、权重更新等操作。训练时需要注意过拟合的问题,并及时进行模型的调整。 7. 模型测试:在训练完成后,使用测试集数据对模型进行测试,以验证模型的准确性和鲁棒性。 总之,mmdetection训练Faster R-CNN需要进行许多步骤,其中需要注意的是数据的预处理、模型搭建和调整超参数、数据扩增、损失函数的设定、并行计算和过拟合的问题。通过对这些步骤进行科学、合理的设置,可以获得较好的结果。 ### 回答3: mmdetection是一个开源的目标检测框架,其中包含了许多常用的检测算法如Faster R-CNN,RetinaNet等模型,同时也提供了细粒度的API接口方便用户进行自定义的检测模型开发。 在使用mmdetection训练Faster R-CNN模型时,首先需要准备数据集。一般而言,数据集应包含包含训练图片、标注文件和验证图片。其中标注文件指的是通过标注工具标记每张图片中要检测的区域,并且标注的格式应符合COCO格式。 接着需要进行模型配置,mmdetection支持许多的配置项,如数据增强方式、anchor设计和ROI pooling等。用户可以根据自己的需求进行设置。对于Faster R-CNN模型而言,它一般需要设置backbone、RPN以及ROI pooling等模块的参数。 在模型配置完成后,就可以使用mmdetection提供的训练函数进行模型训练了。通常使用的是SGD或Adam等优化器,训练时可进行多次迭代,不断优化模型的参数。 另外,模型训练过程中还需要进行模型验证。通常在每一次训练迭代结束后,对于验证集上的表现进行评估,了解模型的效果。在评估过程中,可以计算模型的精度、召回率等指标,据此判断模型的性能是否达到要求,并进行参数的微调。 需要注意的是,Faster R-CNN模型的训练所需要的时间和硬件资源比较大,因此在进行模型训练时,可选择使用GPU进行加速。此外,也需要耐心地对模型进行调整和优化,以提高模型的检测效果。 总的来说,mmdetection训练Faster R-CNN模型需要进行数据集准备、模型配置、模型训练和模型验证等步骤。需要注意的是,模型训练时间较长,需要耐心对模型进行调整和优化,保证最终的检测效果达到要求。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值