栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Ubuntu 18.04 CUDA10.2版本下安装MMDetection笔记 亲测有效哦~

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Ubuntu 18.04 CUDA10.2版本下安装MMDetection笔记 亲测有效哦~

1 预处理

1.1 配置要求

MMCV(相匹配的版本如下所示)

MMCV:Compatible MMDetection and MMCV versions are shown as below. Please install the correct version of MMCV to avoid installation issues

 

 Note: You need to run pip uninstall mmcv first if you have mmcv installed. If mmcv and mmcv-full are both installed, there will be ModuleNotFoundError.(如果原来有mmcv的话先把原来的删掉)

1.2 查看指令

2 安装步骤

2.1 新建环境

指令

①	conda create -n open-mmlab python=3.7 -y #新建虚拟环境“open-mmlab”
②	conda activate open-mmlab # 或 source activate open-mmlab #进入虚拟环境

2.2 安装pytorch

2.2.1 pytorch官网:Start Locally | PyTorch(官网查看对应版本)

2.2.2 相关指令(下面是博主自己对应的版本)

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

2.3 安装mmcv-full(新版一定要安装mmcv-full)

GitHub - open-mmlab/mmcv: OpenMMLab Computer Vision Foundationhttps://github.com/open-mmlab/mmcv#installation(官网查看对应的版本)

安装指令:(这也是博主自己的对应版本,不一定适用)

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html

成功

2.4 安装MMDetection

2.4.1 可以使用以下命令简单地安装MMDetection

pip install mmdet

 2.4.2 或者使用克隆存储库,然后安装

①	git clone https://github.com/open-mmlab/mmdetection.git

 这里可能会报错,通常由于服务器版本问题与无vpn连接GitHub速度缓慢造成。

建议改为:

git clone git://github.com/open-mmlab/mmdetection.git

 

 (下载速度异常缓慢,建议听会儿歌)搞定!

②	cd mmdetection
③	pip install -r requirements/build.txt

④	 pip install -v -e .  # or "python setup.py develop"(成功)

3 测试部分

3.1 新建文件夹

在mmdetection下新建checkpoints文件夹

指令:mkdir checkpoints

在mmdetection/configs目录下有很多个模型,选择其中一种模型的文件夹进去,找到README.md下载对应的模型文件,放到checkpoints文件夹内。这里选择mask_rcnn_r50_fpn_1x_coco_20200205-d4b0c5d6.pth

输入指令:
python demo/image_demo.py demo/demo.jpg configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py checkpoints/mask_rcnn_r50_fpn_1x_coco_20200205-d4b0c5d6.pth(测试成功)

 OK,大功告成,拜拜~

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/649899.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号