mmdetection环境搭建

本文详细介绍了在Ubuntu16.04环境下,使用Anaconda3搭建mmdetection实例分割工具箱的过程,包括虚拟环境创建、依赖包安装、源配置、mmcv与mmdetection的克隆与安装等关键步骤。

 

最近从语义分割转做实例分割,需要用到mmdetection,作为从来没有用过检测工具箱的我,趟了各种坑之后特写此博客供后来者参考(这篇文章算是集众家之所长吧)。


系统环境:

ubuntu16.04

cuda:10.0(此处一定注意检查自己的环境,关乎到后面cudatoolkit版本的安装问题)

1.安装anaconda3,这一步不讲,很简单,去官网下载文件运行安装就好了。

2.创建open-mmlab的虚拟环境并激活;

conda create -n open-mmlab python=3.7
conda activate open-mmlab

3.为了避免conda下载包速度慢,先添加官方及第三方清华镜像源(这个地方注意一下,是http,http,http,不是别的博客说的https,写https能报HTTP ERROR报到你崩溃,趟了无数遍坑之后血和泪的教训);

​conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

 4.接下来开始搭建环境;

先安装numpy1.16.0,因为此时不安装后面会自动安装numpy1.18.2,numpy的版本会直接关乎到最后模型测试时能否正常计算map指标。

conda install numpy==1.16.0

安装pytorch

conda install pytorch torchvision cudatoolkit=10.0

 注意此处的cudatoolkit=10.0是对应你的cuda版本的,如果你的cuda版本是9.0/9.2/10.1,那么相应改为cudatoolkit=9.0/9.2/10.1,不知道自己的cuda版本?去/usr/local/下去看。

安装cython

conda install cython

安装mmcv

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .(是的你没看错,这里就是有个 . )

安装mmdetection

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
python setup.py develop

 安装pycocotools

conda config --add channels conda-forge
conda install pycocotools

至此,mmdetection环境安装完成。

### mmdetection 环境搭建教程及使用指南 #### 三、mmdetection环境配置流程 为了顺利运行mmdetection,推荐通过Anaconda3创建新的虚拟环境来管理依赖项[^1]。 ```bash conda create -n open-mmlab python=3.8 -y conda activate open-mmlab ``` 接着安装必要的Python包: ```bash pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10/index.html pip install mmdet ``` 上述命令会自动处理所有必需的依赖关系并完成软件包的安装。对于CUDA版本的选择,请依据个人硬件情况调整URL中的`cu113`部分以匹配所使用的NVIDIA驱动程序版本;同理,根据PyTorch版本更改`torch1.10`这部分内容[^4]。 针对特定错误如`/lib/python3.7/site-packages/mim/commands/search.py`文件第392行抛出异常的情况,在遇到此类问题时可以尝试按照建议的方式修改代码逻辑,即将条件判断语句改为 `if collection_name and collection_name in name2collection.keys():` 来解决潜在的问题[^5]。 #### 四、mmdetection 使用入门 成功设置好开发环境之后,可以通过如下方式验证安装是否正确无误: ```python from mmdet.apis import init_detector, inference_detector import mmcv config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth' model = init_detector(config_file, checkpoint_file, device='cuda:0') img = mmcv.imread('demo/demo.jpg') result = inference_detector(model, img) mmcv.imshow_det_bboxes(img, result, model.CLASSES, score_thr=0.5) ``` 这段脚本展示了如何加载预训练模型并对单张图片执行目标检测任务。用户可以根据实际需求替换配置文件路径(`config_file`)和权重文件路径(`checkpoint_file`),同时也可以指定待测试图像的位置。
评论 4
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值