做大做强系列:mmdetection(一) 环境配置

做大做强系列:mmdetection(一) 环境配置

下篇讲训练自己的数据集和调参。

一说到环境配置,我想和大多数宝宝一样,打开百度或者google,输入关键词“mmdetetion安装…”或者“mmdetection训练自建的数据集…”,当然网上的安装的教程翻来翻去大同小异,再看这些教程blog下的评论区各位宝宝依旧有各种各样的问题。

“我这篇教程基本上会涵盖各位宝宝的各种各样的问题”,当然不能!

但是按照我的采坑经验,给大家真心建议不要看各种第三方的环境配置文档,等我说完下句,各位宝宝就可以退出来,重点!!!请参考官方给的安装文档进行安装(https://mmdetection.readthedocs.io/en/latest/get_started.html#installation),好了,各位宝宝可以退出了。那我接下来写什么呢?当然是记录一下安装过程(翻译官方文档加更新),不想看英文的宝宝可以留下来看我的,嘻嘻。

1、创建虚拟环境(和官方文档一致)
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab
2.安装torch、torchvision(不推荐官方文档给出的安装方式,安装速度太慢了)
注意:确保CUDA版本和torch的匹配。 您可以在PyTorch网站(https://pytorch.org/)上查找支持的不同CUDA版本的pytorch。
pip install torch版本号+cu版本 torchvision版本号+cu版本 -f https://download.pytorch.org/whl/torch_stable.html
例如:
pip install torch1.5.0+cu92 torchvision0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
3.安装mmcv(版本需要和torch吻合):
官方文档中给了三种安装方式,怕麻烦的宝宝,直接输入:
pip install mmcv-full
4.下载mmdetection(官网更新很快,git clone最新版本即可):
这里有个小trick,可以加快git clone的速度(加中国域名):
git clone https://github.com.cnpmjs.org/open-mmlab/mmdetection.git
5、安装编译mmdetection:
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e . # or “python setup.py develop”
6、测试环境是否可用:
python tools/train.py -h 出现下图所示的结果,表示安装成功在这里插入图片描述

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值