安装低版本MMDetection稳定运行,一次安装成功

最新的MMDetection版本可能在一些机器上运行不起来,主要是里面的库mmcv>2.0mmdet>3.0与cuda不适配,这里推荐安装mmcv<2.0mmdet<3.0,MMDetection可以稳定运行。

1.创建环境
conda create -n mmlab python=3.8 -y
2.进入环境
conda activate mmlab
3.安装低版本的 pytorch, cuda
pip install torch==1.13.0 torchvision --index-url https://download.pytorch.org/whl/cu117
4. 安装mmcv

低版本中的mmcv使用关键字,mmcv-full,这里的是mmcv1.7.0

安装 MMCV — mmcv 1.7.1 文档

pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html
5. 安装mmdet

选择要安装的低版本mmdet,进入网站,这里选择v2.26.0版本,点击下载安装包

Tags · open-mmlab/mmdetection · GitHub

下载好,在conda环境中进入文件夹,安装必要的库

pip install -v -e . # 句尾有个点别忽略了

安装完成后,检查一下所安装的库

6. 安装测试

下载权重 下载权重

填写文件夹 demo/image_demo.py关键参数

# demo/ 文件夹下面demo.jpu
parser.add_argument('--img',default=r'F:\openmmlab\mmdetection-2.26.0\demo\demo.jpg', help='Image file')
# 使用这个算法文件进行安装测试
parser.add_argument('--config',default=r'F:\openmmlab\mmdetection-2.26.0\configs\faster_rcnn\faster_rcnn_r50_fpn_1x_coco.py', help='Config file')
# 已下载的权重,新建checkpoints文件夹,放在里面
parser.add_argument('--checkpoint', default=r'F:\openmmlab\mmdetection-2.26.0\checkpoints\faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth', help='Checkpoint file')

 关键参数如下所示:

# 运行image_demo.py进行测试
python demo/image_demo.py

 测试结果如下所示:

低版本的MMDetection安装成功!

以上是Windows安装步骤,Linux Ubuntu系统安装同理~

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值